Convert fast.ai trained image classification model to iOS app via ONNX and Apple Core ML

Flow

Pytorch → ONNX → Apple Core ML

What we need to install

Steps

Replace AdaptivePooling with fixed size Pooling
Extract all layers with few modifications
cat
dog
Convert Pytorch → onnx → Apple Core ML

--

--

--

Love podcasts or audiobooks? Learn on the go with our new app.

Recommended from Medium

Tanimoto VS. Mol2vec

Artificial Neural Network vs Deep Neural Network

Anomaly Detection on the Surface of Mars.

Is PP-YOLOE Better than YOLOv5 & YOLOX?

Quick Breakdown on Supervised Machine Learning

Dog Classifier Project

Face Detection using MTCNN

Music Album Cover Generation from Music Track

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
Hung Minh Nguyen

Hung Minh Nguyen

More from Medium

OpenVINO Pre-trained Model Application Demos

FM Logistic speeds up document processing with LearningPal, a Silicon Valley based startup.

gRPC rocks build your first gRPC service(part 1)

IP Insights Model ‘De-simplify’ — Part III Model Training