Toybrick

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

ldol31627

中级会员

积分
310
楼主
发表于 2019-4-9 14:28:01    查看: 38696|回复: 11 | [复制链接]    打印 | 只看该作者
本帖最后由 ldol31627 于 2019-4-9 14:30 编辑

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

使用道具 举报

ldol31627

中级会员

积分
310
沙发
 楼主| 发表于 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

中级会员

积分
452
板凳
发表于 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

中级会员

积分
396
地板
发表于 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

版主

积分
894
5#
发表于 2020-3-10 09:37:09 | 只看该作者
Devin 发表于 2020-2-14 17:07
请问下,如果是caffe训练的mobilenet ssd怎么生成priorbox.txt文件呢?谢谢!!

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

使用道具 举报

leok

版主

积分
894
6#
发表于 2020-3-10 09:51:16 | 只看该作者
Ryan_zdst 发表于 2020-3-2 09:08
这个在tensorflow  object detection api 中运行不了啊

可以去object_detection_tutorial.ipynb里头跑
回复

使用道具 举报

ylc123

注册会员

积分
146
7#
发表于 2020-6-2 11:52:16 | 只看该作者
本帖最后由 ylc123 于 2020-6-3 15:00 编辑
leok 发表于 2020-3-10 09:37
找到concat节点,输出按格式保存。

已解决。
回复

使用道具 举报

kitedream

中级会员

积分
284
8#
发表于 2020-7-16 17:26:52 | 只看该作者
leok 发表于 2020-3-10 09:51
可以去object_detection_tutorial.ipynb里头跑

你好,我在object_detection_tutorial.ipynb目前可以调用模型,正常输出结果。也在ensorflow\models\research\object_detection\meta_architectures\ssd_meta_arch.py的相应位置添加了如下代码,但整个推理过程结束后,也没有生成文件。请问,是不是我修改位置错了?



在 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')
回复

使用道具 举报

anqingcheng

注册会员

积分
73
9#
发表于 2020-12-15 09:03:32 | 只看该作者
darknet  yolov3  怎么生成priorbox.txt文件?
回复

使用道具 举报

zehao

注册会员

积分
166
10#
发表于 2021-3-1 11:34:37 | 只看该作者
kitedream 发表于 2020-7-16 17:26
你好,我在object_detection_tutorial.ipynb目前可以调用模型,正常输出结果。也在ensorflow\models\rese ...

请问最后这个问题解决了吗,在ipynb里面可以生成对应的box prior了吗
回复

使用道具 举报

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

本版积分规则

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


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