ppogg/YOLOv5-Lite

lite-g版本输出跟其他版本不一样

liuxufenfeiya opened this issue · 1 comments

其他都是3个输出,比如c,output,1111,2222
但是g版本的,输出只有output,看结构是做了concat,
但是直接拿取上层的输出485,513,541这三个,用原来的代码,也无法输出正确的结果(在安卓侧,使用ncnn版本转换onnx模型)

ppogg commented

如果使用export.py,记得吧--concat设置为False,这样可以把分支检测头解掉