onnx / onnx

Open standard for machine learning interoperability
https://onnx.ai/
Apache License 2.0
17.35k stars 3.64k forks source link

Does anyone knows the preprocessing and postprocessing of YOLOv2 coco from onnx zoo? #5533

Open Apisteftos opened 11 months ago

Apisteftos commented 11 months ago

I found a code about the tiny yolov2 onnx and followed the instructions from C# and darknet about the anchors and preprocessing.

I am using the

anchors = [0.57273, 0.677385, 1.87446, 2.06253, 3.33843, 5.47434, 7.88282, 3.52778, 9.77052, 9.16828]

tooked from darknet. Now the postprocessing should be similar to tiny yolov2 which seems isn't.

I am getting wrong class prediction + the bounding boxes are not in place.

xadupre commented 11 months ago

Is it possible to know more about the models you are comparing? (location, converter you used, ...)

Apisteftos commented 11 months ago

Is it possible to know more about the models you are comparing? (location, converter you used, ...)

I am not converting them, I am using these from the onnx zoo. I am doing my thesis in AI accelerators and testing the performance of the pre-trained onnx models from onnx zoo. I did that for almost all classification models and couple of them of the object detections & image segmentation, but they have huge problems.

Problems:

Tested from Object Detection: