search

onnx / models

A collection of pre-trained, state-of-the-art models in the ONNX format
http://onnx.ai/models/
Apache License 2.0
7.97k stars 1.41k forks source link

At Readme of YOLOv3, Value types of original image size need to be float #671

Open susumu-iino opened 2 months ago

susumu-iino commented 2 months ago

Bug Report

Which model does this pertain to?

YoloV3

Describe the bug

Each of hosted models(YOLOv3, YOLOv3-12, YOLOv3-12-int8) need the original image size as the second input. These Onnx Models expect float values for the second input, but the preprocessing steps sets it as dtype=np.int32, this causes asserting by type error. It need to be np.float32 or 'float32' for the parameter dtype on this last line, same as Tiny Yolov3.

I will make a pull request for this issue soon.

Reproduction instructions

System Information

OS Platform and Distribution:
Windows 11 22H2

ONNX version:
not needed for reproducing this issue

Backend/Runtime version: onnxruntime 1.18.0 numpy 1.26.0