ssd模型是基于哪个版本转的.rknn？

[复制链接] · 发表于 2020-1-9 17:47:27

官方在3399pro上ssd demo里面的.rknn模型是基于哪个版本转的，可以给个链接吗，如果只说官方的话有歧义，是tf的官方还是哪里的官方，不太能理解，谢谢大佬们

只看该作者 · 发表于 2020-1-10 08:58:48

https://github.com/tensorflow/mo ... ection_model_zoo.md

只看该作者 · 发表于 2020-1-14 09:28:06

请问一下：就是在转换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？

只看该作者 · 发表于 2020-3-20 16:27:29

xupeng 发表于 2020-1-14 09:28
请问一下：就是在转换tensorflow(object_detection这个tf官方的检测框架) ssd_resnet50_fpn_pb 到 rknn模型 ...

请问下，你是直接下载的模型，还是自己训练的

只看该作者 · 发表于 2020-3-21 16:03:40

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帖子