Tianxiaomo / pytorch-YOLOv4

PyTorch ,ONNX and TensorRT implementation of YOLOv4
Apache License 2.0
4.46k stars 1.49k forks source link

Output ONNX file #573

Open chingkhei-th opened 3 months ago

chingkhei-th commented 3 months ago

I was trying to convert my tiny-yolov4 darknet weight into ONNX format. When I run python demo_darknet2onnx.py <cfgFile> <namesFile> <weightFile> <imageFile> <batchSize> by replacing each values. I got 2 ONNX model file:

Is this correct? I was expecting some output like custom_yolov4_tiny.onnx. I'm not sure if my output model is in tiny yolov4 format or just regular yolov4 format. Also it will be really helpful if someone can suggest me how can I use my ONNX model in Unity.

Thanks in advance!