pytorch-YOLOv4
pytorch-YOLOv4 copied to clipboard
Output ONNX file
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:
- yolov4_1_3_416_416_static.onnx
- yolov4_64_3_416_416_static.onnx
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!