建议使用以下浏览器,以获得最佳体验。 IE 9.0+以上版本 Chrome 31+ 谷歌浏览器 Firefox 30+ 火狐浏览器
请选择 进入手机版 | 继续访问电脑版
设置昵称

在此一键设置昵称,即可参与社区互动!

确定
我再想想
选择版块

gcl

发帖: 2粉丝: 0

级别 : 新手上路

Rank: 1

发消息 + 关注

发表于5 天前 103 1 楼主 显示全部楼层
[问题求助] 关于官方给的sample-objection与sample-objectdetectionbyyolov3的代码的困惑

sample-objection中给出的模型推理代码如图:

image.png

输入信息中需要包含img_info. 但我在sample-objectdetectionbyyolov3中发现yolov3输入信息不需要包含img_info:

image.png

我对sample-objection进行了测试,发现如果不包含img_info,模型在推理阶段会报错,如图:

3.png

我想问一下,这是为什么。

举报
分享

分享文章到朋友圈

分享文章到微博

windyxyz

发帖: 12粉丝: 14

级别 : 高级会员

Rank: 4

发消息 + 关注

发表于5 天前 沙发 显示全部楼层

你好,查看下两个模型的prototxt文件,fasterrcnn的input 有两个节点

input: "data"
input_shape {
  dim: 1
  dim: 3
  dim: 600
  dim: 800
}

input: "im_info"
input_shape {
  dim: 1
  dim: 3

} 而yolov3的input 有一个节点:

input: "data"
input_dim: 1
input_dim: 3
input_dim: 416

input_dim: 416 在转om模型时,fasterrcnn 有两个输入,yolov3只有一个输入。推理输入数据和模型要求的输入数据需要保持一致 , 所以物体检测 input_data_vec  push了两个tensor,而yolov3 input_data_vec  push了一个tensor。


点赞 评论 引用 举报

游客

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