TexasInstruments / edgeai-yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite. Forked from https://ultralytics.com/yolov5

Home Page:https://github.com/TexasInstruments/edgeai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

how could i get the prototxt file when export yolo-pose model?

happyamyhope opened this issue · comments

❔Question

how could i get the prototxt file when export yolo-pose model?

Additional context

when i use export.py as below,
python models/export.py --weights runs/train/exp/weights/best.pt --img 640 --batch 1 --simplify --export-nms # export at 640x640 with batch size 1
only get the onnx model file and torchscript model file, without prototxt file,and even i add export_prototxt func, there is something error.