Skip to content

example of Core ML Tools when converted from TensorFlow and PyTorch

Notifications You must be signed in to change notification settings

Shakshi3104/Convert2CoreMLDemo

Repository files navigation

Convert to Core ML Demo

This repository shows example code for the same Core ML model when converted from TensorFlow and when converted from PyTorch.

Requirements

  • coremltools 4.1
  • tensorflow 2.1
  • torch 1.6
  • torchvision 0.7

See requirement.txt for all required packages

Example of Unified Conversion API - MobileNet

Example of converting pre-trained MobileNet v2 from TensorFlow and PyTorch.

Code is here.

About

example of Core ML Tools when converted from TensorFlow and PyTorch

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published