Skip to content

Instantly share code, notes, and snippets.

@jknthn

jknthn/model_conversion.py

Created June 22, 2017 10:45
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jknthn/c665c5d253dc6d8e460ac8061fbdb894 to your computer and use it in GitHub Desktop.
Save jknthn/c665c5d253dc6d8e460ac8061fbdb894 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model_conversion.py
import coremltools
age_net = coremltools.converters.caffe.convert(('models/age_net.caffemodel', 'models/deploy_age.prototxt'),
image_input_names='data',
class_labels='models/age_labels.txt')
gender_net = coremltools.converters.caffe.convert(('models/gender_net.caffemodel', 'models/deploy_gender.prototxt'),
image_input_names='data',
class_labels='models/gender_labels.txt')
age_net.save('AgeNet.mlmodel')
gender_net.save('GenderNet.mlmodel')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment