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文件呢?谢谢!!




欢迎光临 Toybrick (http://t.rock-chips.com/) Powered by Discuz! X3.3