Core ML モデルのメタデータを編集する

変換元のリポジトリとかライセンスとか書くのに便利

import coremltools as ct

model =  ct.models.MLModel('LaMa.mlpackage')

# Set the model metadata
model.author = 'https://github.com/advimman/lama'
model.license = 'Apache2.0 License'
model.short_description = 'LaMa: Resolution-robust Large Mask Inpainting with Fourier Convolutions.'


# Get the interface to the model
model.input_description
model.output_description

# # Set feature descriptions manually
# model.input_description['input'] = "input image of scene to be classified"
# model.output_description['output'] = "Probability of each category"

# Save the model
model.save('lama.mlpackage')

🐣

---

フリーランスエンジニアです。
AIについて色々記事を書いていますのでよかったらプロフィールを見てみてください。

もし以下のようなご要望をお持ちでしたらお気軽にご相談ください。
AIサービスを開発したい、ビジネスにAIを組み込んで効率化したい、AIを使ったスマホアプリを開発したい、
ARを使ったアプリケーションを作りたい、スマホアプリを作りたいけどどこに相談したらいいかわからない…

いずれも中間コストを省いたリーズナブルな価格でお請けできます。

お仕事のご相談はこちらまで
[e-mail]

機械学習やAR技術を使ったアプリケーションを作っています。
機械学習／AR関連の情報を発信しています。

X
Medium
GitHub