issues
search
Ziyu98
/
my_test_on_TNN_yolov3
2
stars
0
forks
source link
请教一下,你的yolov3模型转tnn的时候,做了些什么,修改
#1
Open
cncowboy
opened
4 years ago
cncowboy
commented
4 years ago
ziyu你好,我也在尝试在tnn上跑yolov3做目标检测,但是在转换时,报错了, 所以想请教你做了哪些修改,能跑通。
Ziyu98
commented
4 years ago
yolov3-->onnx 我的yolov3用的是pytorch,具体是
https://github.com/eriklindernoren/PyTorch-YOLOv3
,这一步我把model中的yolo层删掉了,让模型的输出为三个feature map(255
13
13, 255
26
26, 255
52
52);同时改了一下config文件中的'route'层(这个是为了确保route有两个输入参数)
onnx-->tnn 这个就直接转换就好,我没做更改
cncowboy
commented
4 years ago
感谢感谢。
ziyu你好,我也在尝试在tnn上跑yolov3做目标检测,但是在转换时,报错了, 所以想请教你做了哪些修改,能跑通。