ObjectDetection-CoreML

Requirements

How To Build and Run the Project

1. Clone the project

git clone https://github.com/tucan9389/ObjectDetection-CoreML

2. Prepare Core ML model

3. Add the model to the project

4. Set model name properly in ViewController.swift

5. Build and Run

How To Run with your own model

1. Convert your model to Core ML

Pre-requirements

pip install ultralytics
pip install coremltools

Option 1) With shell

yolo export model=yolov8n.pt format=coreml nms

Option 2) With python script

# mian.py
from ultralytics import YOLO
if __name__ == '__main__':
 model = YOLO("yolov8n.pt", task='detect') # load a pretrained model
 model.overrides['nms'] = True
 success = model.export(format="coreml") # export the model to CoreML format

# in terminal
python main.py
# then you can see the `.mlpackage` or `.mlmodel` file in your current directory
# (btw you can check your current directory with `pwd` command)

2. Follow the steps above from Step 3

Models

Model Matadata

Model Size, Minimum iOS Version, Download Link

Trained Dataset Infos

Performance

Infernece Time (ms)

Total Time (ms)

FPS

See also