pytorch-YOLOv4 icon indicating copy to clipboard operation
pytorch-YOLOv4 copied to clipboard

Output ONNX file

Open chingkhei-th opened this issue 3 months ago • 0 comments

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!

chingkhei-th avatar Mar 30 '24 13:03 chingkhei-th