Toybrick

标题: rknn-toolkit\example\mobilenet-ssd中的box_priors.txt文件是如何得到的 [打印本页]

作者: ldol31627 时间: 2019-4-9 14:28
标题: rknn-toolkit\example\mobilenet-ssd中的box_priors.txt文件是如何得到的
本帖最后由 ldol31627 于 2019-4-9 14:30 编辑

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

作者: ldol31627 时间: 2019-4-11 13:32
解决方法：
在 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 时间: 2020-2-14 17:07

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 时间: 2020-3-2 09:08

ldol31627 发表于 2019-4-11 13:32
解决方法：
在 tensorflow\models\research\object_detection\meta_architectures\ssd_meta_arch.py
p ...

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

作者: leok 时间: 2020-3-10 09:37

Devin 发表于 2020-2-14 17:07
请问下，如果是caffe训练的mobilenet ssd怎么生成priorbox.txt文件呢？谢谢！！

找到concat节点，输出按格式保存。

作者: leok 时间: 2020-3-10 09:51

Ryan_zdst 发表于 2020-3-2 09:08
这个在tensorflow object detection api 中运行不了啊

可以去object_detection_tutorial.ipynb里头跑

作者: ylc123 时间: 2020-6-2 11:52
本帖最后由 ylc123 于 2020-6-3 15:00 编辑

leok 发表于 2020-3-10 09:37
找到concat节点，输出按格式保存。

已解决。

作者: kitedream 时间: 2020-7-16 17:26

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 时间: 2020-12-15 09:03
darknet yolov3 怎么生成priorbox.txt文件？

作者: zehao 时间: 2021-3-1 11:34

kitedream 发表于 2020-7-16 17:26
你好，我在object_detection_tutorial.ipynb目前可以调用模型，正常输出结果。也在ensorflow\models\rese ...

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

作者: tjmies 时间: 2023-3-30 13:17
你解决了吗

作者: tjmies 时间: 2023-3-30 13:18

anqingcheng 发表于 2020-12-15 09:03
darknet yolov3 怎么生成priorbox.txt文件？

请问最后这个问题解决了吗

欢迎光临 Toybrick (https://t.rock-chips.com/)