Skip to content

junmcenroe/YOLOv5-CoreML-Converter

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

YOLOv5-CoreML-Converter

Recent YOLOv5 export.py do not support train model. For CoreML conversion, export.py need train mode exporting. This export.py can export 3 multiarray without NMS and detect layer. After replacing the original export.py to this, you can convert CoreML model using following scripts

Here is the blog that I followed to run this script:

https://rockyshikoku.medium.com/convert-yolov5-to-coreml-also-add-a-decode-layer-113408b7a848

https://colab.research.google.com/drive/1uR738UTlzI7apqeN0qr6mQ5ke_a5SKa8?usp=sharing

I made YOLOv5CoreMLConverter.py referencing above script. This conveter can get argument

to use this script , to run following

YOLOv5CoreMLConvertProcess.ipynb

YOLOv5CoreMLConverter.py is for YOLOv5 P5 model

YOLOv5P6CoreMLConverter.py is for YOLOv5 P6 1280image size model

About

YOLOv5 CoreML Coverter using export.py with train mode

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published