yolo-fastestv2-opencv
yolo-fastestv2-opencv copied to clipboard
Convert pytorch to onnx
我在原代码model/detector.py的forward function里修改,因为我看这里有两个tensor output,但他们size不一样
我也不是很确定是不是修改这里,求指导~ Any helps will be appreciated!!
Originally posted by @danielzhangau in https://github.com/hpc203/yolo-fastestv2-opencv/issues/1#issuecomment-976087012
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
我在原代码model/detector.py的forward function里修改,因为我看这里有两个tensor output,但他们size不一样
我也不是很确定是不是修改这里,求指导~ Any helps will be appreciated!!
Originally posted by @danielzhangau in #1 (comment)
仓库里已经添加了convert_onnx文件夹,里面有转onnx的程序文件