Toybrick

标题: ssd模型是基于哪个版本转的.rknn? [打印本页]

作者: mrzs    时间: 2020-1-9 17:47
标题: ssd模型是基于哪个版本转的.rknn?
官方在3399pro上ssd demo里面的.rknn模型是基于哪个版本转的,可以给个链接吗,如果只说官方的话有歧义,是tf的官方还是哪里的官方,不太能理解,谢谢大佬们
作者: zht    时间: 2020-1-10 08:58
https://github.com/tensorflow/mo ... ection_model_zoo.md
作者: xupeng    时间: 2020-1-14 09:28
请问一下:就是在转换tensorflow(object_detection这个tf官方的检测框架) ssd_resnet50_fpn_pb 到 rknn模型的时候,输入、输出节点分别是哪些,我目前跑通了ssd-mobilenet_v1,自己生成了box_pirors;但是在跑ssd_resnet50_fpn这个的时候,
我输入节点是/FeatureExtractor/resnet_v1_50/resnet_v1_50/Pad or FeatureExtractor/resnet_v1_50/resnet_v1_50/conv1/Conv2D
输出是 [concat, concat_1]; 都跑的不对。。。
然后你们方便提供你们的输入输出节点吗?以及生成的关于ssd_resnet50_fpn的priors.txt?
作者: lijingle    时间: 2020-3-20 16:27
xupeng 发表于 2020-1-14 09:28
请问一下:就是在转换tensorflow(object_detection这个tf官方的检测框架) ssd_resnet50_fpn_pb 到 rknn模型 ...

请问下,你是直接下载的模型,还是自己训练的
作者: leok    时间: 2020-3-21 16:03
xupeng 发表于 2020-1-14 09:28
请问一下:就是在转换tensorflow(object_detection这个tf官方的检测框架) ssd_resnet50_fpn_pb 到 rknn模型 ...

ssd_resnet50_fpn_pb输入输出:输入节点:FeatureExtractor/resnet_v1_50/resnet_v1_50/conv1/Conv2D 输输出节点:'concat_1', 'concat'     确认过节点是对的。
box_prior生成:参考http://t.rock-chips.com/forum.ph ... 8&highlight=ssd帖子




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