yolo-fastestv2-opencv icon indicating copy to clipboard operation
yolo-fastestv2-opencv copied to clipboard

Convert pytorch to onnx

Open danielzhangau opened this issue 2 years ago • 2 comments

我在原代码model/detector.py的forward function里修改,因为我看这里有两个tensor output,但他们size不一样 image

image 我也不是很确定是不是修改这里,求指导~ Any helps will be appreciated!!

Originally posted by @danielzhangau in https://github.com/hpc203/yolo-fastestv2-opencv/issues/1#issuecomment-976087012

danielzhangau avatar Nov 25 '21 06:11 danielzhangau

issue fixed! Can check out my new pull request in the original work. Hope it helps! https://github.com/dog-qiuqiu/Yolo-FastestV2/pull/44#issue-1064116145

danielzhangau avatar Nov 26 '21 05:11 danielzhangau

我在原代码model/detector.py的forward function里修改,因为我看这里有两个tensor output,但他们size不一样 image

image 我也不是很确定是不是修改这里,求指导~ Any helps will be appreciated!!

Originally posted by @danielzhangau in #1 (comment)

仓库里已经添加了convert_onnx文件夹,里面有转onnx的程序文件

hpc203 avatar Dec 04 '21 08:12 hpc203