Exporting from yolov5 to coreml #8
Description
having trouble exporting our ultralytics yolov5 models to coreml.
git clone repo, cd repo, checkout fix/coreml_export_nms_layer, install req, run export cmd
python3 export.py --weights ../../output/best.pt --include coreml
... lots of stuff happening, and eventually ...
added const 10 for builder <coremltools.models.neural_network.builder.NeuralNetworkBuilder object at 0x7f078bb94580>
added const 11 for builder <coremltools.models.neural_network.builder.NeuralNetworkBuilder object at 0x7f078bb94580>
added const 12 for builder <coremltools.models.neural_network.builder.NeuralNetworkBuilder object at 0x7f078bb94580>
Translating MIL ==> NeuralNetwork Ops: 97%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▋ | 656/678 [00:00<00:00, 1012.99 ops/s]added const 878 for builder <coremltools.models.neural_network.builder.NeuralNetworkBuilder object at 0x7f078bb94580>
Translating MIL ==> NeuralNetwork Ops: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 678/678 [00:00<00:00, 1160.80 ops/s]
CoreML: export failure: 0 has type int, but expected one of: bytes, unicode
/usr/src/app/temp/yolov5#
if its helpful we can share our best.pt.