Skip to content

Instantly share code, notes, and snippets.

@sgl0v

sgl0v/image_colorizer_alpha_version_convert.py

Created June 2, 2021 19:16
Show Gist options
  • Save sgl0v/4590039a836296150bf0ac10c7b62316 to your computer and use it in GitHub Desktop.
Save sgl0v/4590039a836296150bf0ac10c7b62316 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
image_colorizer_alpha_version_convert.py
def convert_model(model):
model.save('tf_keras_model')
mlmodel = ct.convert('tf_keras_model')
mlmodel.save("model_alpha.mlmodel")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment