YOLO-CoreML-Converter

Introduction

A conversion tool to convert Weights file, TensorFlow and PyTorch model to CoreML model via coremltools. Inspired by MPieter.

Weights files

weight_file_converter: A converter to convert Weights file to CoreML model.

• Support both YOLOv3, Tiny YOLOv3, YOLOv4 and Tiny YOLOv4 models
• Support to export Keras h5 format model

Other models

unified_converter: A converter to convert TensorFlow and PyTorch model to CoreML model by using coremltools unified Conversion API.

• Support any model supported by coremltools

---

Pretrained Models

Weights file Models

• YOLOv3
  Weights file: https://pjreddie.com/media/files/yolov3.weights
  Configure file: https://raw.githubusercontent.com/AlexeyAB/darknet/master/cfg/yolov3.cfg
• Tiny YOLOv3
  Weights file: https://pjreddie.com/media/files/yolov3-tiny.weights Configure file: https://raw.githubusercontent.com/AlexeyAB/darknet/master/cfg/yolov3-tiny.cfg
• YOLOv4
  Weights file: https://drive.google.com/open?id=1bV4RyU_-PNB78G-OtoTmw1Q7t_q90GKY.
  If above link cannot work, please go to (YOLOv4-model-zoo)[https://github.com/AlexeyAB/darknet/wiki/YOLOv4-model-zoo] and download YOLOv4-Mish-416 or YOLOv4-Leaky-416 weights file.
  Configure file: https://raw.githubusercontent.com/AlexeyAB/darknet/master/cfg/yolov4.cfg
• Tiny YOLOv4
  Weights file: https://github.com/AlexeyAB/darknet/releases/download/yolov4/yolov4-tiny.weights If above link canot work, please go to https://github.com/AlexeyAB/darknet/releases/tag/yolov4 and download file yolov4-tiny.weights.
  Configure file: https://raw.githubusercontent.com/AlexeyAB/darknet/master/cfg/yolov4-tiny.cfg

Keras H5 models

https://github.com/david8862/keras-YOLOv3-model-set

Quick Start

Weights files

Download the pretrained model and run below command.

python weight_file_converter.py yolov4.cfg yolov4.weights model_data/yolov4.mlmodel --keras_output_path model_data/yolov4.h5

To also export the intermediate Keras model use the argument --keras_output_path.

Other models

python unified_converter.py yolov4-tiny-416 fine_tuned.mlmldel -t

For YOLO tiny model, please specify -t option.

Tested with

- Python 3.x.x
- Keras 2.2.4
- tensorflow 2.3.0
- coremltools 5.0

Verify Converted Models

To verify the converted YOLO model, please use my YOLO CoreML project to verify.