ncnn icon indicating copy to clipboard operation
ncnn copied to clipboard

Wrong result for onnx-ncnn yolov5s_6.0

Open Draft00 opened this issue 2 years ago • 7 comments

Problem: incorrect detection

Windows platform. Image with wrong detection for example: image

model | 模型 | モデル

  1. original model: https://github.com/ultralytics/yolov5

how to reproduce | 复现步骤 | 再現方法

  1. Install yolov5 steps from Quick Start Examples.
  2. Verify: python detect.py --source data/images --weights yolov5s.pt --conf 0.25.
  3. Export for solve problems 'Unsupported slice step !': python export.py --include onnx --weight yolov5s.pt --img 640 --train --simplify. According to this.
  4. Conversion using out-of-the-box web model conversion (https://convertmodel.com/#outputFormat=ncnn): check both 'Optimize', check 'Generate ft16 model'.
  5. Comparing my .param file with the param file example (https://github.com/nihui/ncnn-assets/blob/master/models/yolov5s_6.0.param). The layers are the same, only the numbers differ. Size the .bin file is the same as in the example
  6. Use ncnn tutorial (https://github.com/Tencent/ncnn/blob/master/examples/yolov5.cpp) for yolov5s_6.0. Rewrite ex.extract() according to my .param file numbers.
  7. ...And I get a completely incorrect detection result :(

@zhiliu6, could you tell me how you converted the model? it seemed to me that I was doing similar steps, but this is not the case :(

Draft00 avatar Mar 07 '22 14:03 Draft00

I have similar problems,run on android,params yololv5 6.0 by pnnx

traffic_model.zip

xbc922 avatar Mar 26 '22 18:03 xbc922

Running into a similar issue. Have you had any luck resolving this @xbc922 @Draft00 @zhiliu6?

lizozom avatar May 26 '22 12:05 lizozom

Running into a similar issue. Have you had any luck resolving this @xbc922 @Draft00 @zhiliu6?

@lizozom, Nope :(. I had to use yolov4.

Draft00 avatar May 26 '22 13:05 Draft00

@Draft00 my intuition tells me that the processing of images is different between the python and JavaScript versions. I'm not an ML pro, so I'm having a hard time analysing this myself.

lizozom avatar May 29 '22 16:05 lizozom

I have the same issue with yolov5n.

houshoubao avatar Jun 09 '22 08:06 houshoubao

cc I ended up using an older version as well.

lizozom avatar Sep 05 '22 14:09 lizozom

the same error, can you solve it ?

nqthai309 avatar Sep 16 '22 02:09 nqthai309