zherlock030 / YOLOv5_Torchscript

C++ code for running a yolov5s model.
28 stars 5 forks source link

no objects, line 401 #2

Open pengyang1225 opened 4 years ago

pengyang1225 commented 4 years ago

I converted the latest onnx and pt ,But it did not detect the target。can you give me export_onnx.py or .onnx

zherlock030 commented 4 years ago

@pengyang1225 hey, sorry that I did't try onnx, but I think it is included in export.py. when u export, let the detect layer output the inputed list 'x'.

pengyang1225 commented 4 years ago

can you tell me the onnx shape? thanks