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
  1. yolov3-->onnx 我的yolov3用的是pytorch,具体是 https://github.com/eriklindernoren/PyTorch-YOLOv3,这一步我把model中的yolo层删掉了,让模型的输出为三个feature map(2551313, 2552626, 2555252);同时改了一下config文件中的'route'层(这个是为了确保route有两个输入参数)
  2. onnx-->tnn 这个就直接转换就好,我没做更改
cncowboy commented 4 years ago

感谢感谢。