zherlock030 / YOLOv5_Torchscript

C++ code for running a yolov5s model.
27 stars 6 forks source link

anchor_grid.txt 是怎么得到的,能简单说下么,还有你对export.py改了啥么,还是用的原版的,谢谢老铁了 #3

Open JohnTang93 opened 4 years ago

zherlock030 commented 4 years ago

export.py没有修改,导出时,把detect layer的输出改成输入的x,是个列表。 anchor grid是把python里面的anchor grid打印出来。

JohnTang93 commented 4 years ago

哦哦,了解了,trace了之后只能固定尺寸输入了吧?

zherlock030 commented 4 years ago

@JohnTang93 images in different shapes can be fed. 可以用不同尺寸输入。