Toybrick

rknn-toolkit\example\mobilenet-ssd中的box_priors.txt文件是如何得到的

ldol31627

中级会员

积分
296
发表于 2019-4-9 14:28:01    查看: 2096|回复: 6 | [复制链接]    打印 | 显示全部楼层
本帖最后由 ldol31627 于 2019-4-9 14:30 编辑

请问rknn-toolkit\example\mobilenet-ssd中的box_priors.txt文件是如何得到的?
我的mobilenet-ssd 输入是 640x360, 需要重新定义box_priors.txt文件。
我是用tensorflow训练的模型
回复

使用道具 举报

ldol31627

中级会员

积分
296
 楼主| 发表于 2019-4-11 13:32:55 | 显示全部楼层
解决方法:
在 tensorflow\models\research\object_detection\meta_architectures\ssd_meta_arch.py
    predictions_dict = {
        'preprocessed_inputs': preprocessed_inputs,
        'feature_maps': feature_maps,
        'anchors': self._anchors.get()
    } 代码后添加以下代码:
sess = tf.Session()
    with sess.as_default():
      anchors_np = self._anchors.get().eval()
      anchors_np = anchors_np.transpose()
      minx = anchors_np[0]
      miny = anchors_np[1]
      maxx = anchors_np[2]
      maxy = anchors_np[3]
      cx = (minx + maxx) / 2.
      cy = (miny + maxy) / 2.
      w  = maxx - minx
      h  = maxy - miny
      np.savetxt('box_priors.txt', (cx, cy, w, h), fmt='%0.8f')
即可保存 box_priors.txt 文件
回复

使用道具 举报

Devin

注册会员

积分
199
发表于 2020-2-14 17:07:36 | 显示全部楼层
ldol31627 发表于 2019-4-11 13:32
解决方法:
在 tensorflow\models\research\object_detection\meta_architectures\ssd_meta_arch.py
    p ...

请问下,如果是caffe训练的mobilenet ssd怎么生成priorbox.txt文件呢?谢谢!!
回复

使用道具 举报

Ryan_zdst

注册会员

积分
154
发表于 2020-3-2 09:08:20 | 显示全部楼层
ldol31627 发表于 2019-4-11 13:32
解决方法:
在 tensorflow\models\research\object_detection\meta_architectures\ssd_meta_arch.py
    p ...

这个在tensorflow  object detection api 中运行不了啊
回复

使用道具 举报

leok

版主

积分
595
发表于 2020-3-10 09:37:09 | 显示全部楼层
Devin 发表于 2020-2-14 17:07
请问下,如果是caffe训练的mobilenet ssd怎么生成priorbox.txt文件呢?谢谢!!

找到concat节点,输出按格式保存。
回复

使用道具 举报

leok

版主

积分
595
发表于 2020-3-10 09:51:16 | 显示全部楼层
Ryan_zdst 发表于 2020-3-2 09:08
这个在tensorflow  object detection api 中运行不了啊

可以去object_detection_tutorial.ipynb里头跑
回复

使用道具 举报

ylc123

注册会员

积分
79
发表于 2020-6-2 11:52:16 | 显示全部楼层
本帖最后由 ylc123 于 2020-6-3 15:00 编辑
leok 发表于 2020-3-10 09:37
找到concat节点,输出按格式保存。

已解决。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表