Skip to content

Instantly share code, notes, and snippets.

@gali8

gali8/coremltools_pycharm.py

Last active September 15, 2017 08:14
Show Gist options
  • Save gali8/f2c2a295927cc67cb3c8927e534a14e6 to your computer and use it in GitHub Desktop.
Save gali8/f2c2a295927cc67cb3c8927e534a14e6 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
coremltools_pycharm.py
import coremltools
#https://apple.github.io/coremltools/generated/coremltools.converters.caffe.convert.html
coreml_modelAge = coremltools.converters.caffe.convert(('age_net.caffemodel', 'deploy_age.prototxt', 'mean.binaryproto'), image_input_names='data')
coreml_modelAge.save('Age.mlmodel')
coreml_modelGender = coremltools.converters.caffe.convert(('gender_net.caffemodel', 'deploy_gender.prototxt', 'mean.binaryproto'), image_input_names='data')
coreml_modelGender.save('Gender.mlmodel')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment