Qiita Teams that are logged in
You are not logged in to any team

Log in to Qiita Team
Community
OrganizationAdvent CalendarQiitadon (β)
Service
Qiita JobsQiita ZineQiita Blog
Help us understand the problem. What is going on with this article?

DeepLab V3+ をCore MLに変換する

segmentation.gif
凍結グラフから変換します。

DeepLabをトレーニングする方法、凍結グラフとしてエクスポートする方法は下記の記事を参照していただければ。
DeepLabv3+を自前のデータセットで学習させる

手順

1、必要なライブラリを用意

TensorFlow1をインストールしていない場合はインストール。

pip install tensorflow==1.x

Colabの場合。TensorFlow1を選択。

%tensorflow_version 1.x

Core ML Toolsをインストール。

pip install coremltools==4.0

2、変換

import coremltools as ct

input = ct.ImageType(shape=(1, 513, 513, 3))
mlmodel = ct.convert('./frozen_graph.pb',inputs=[input])
mlmodel.save('./deeplabv3')

🐣


お仕事のご相談こちらまで
rockyshikoku@gmail.com

Core MLを使ったアプリを作っています。
機械学習関連の情報を発信しています。

Twitter
Medium

john-rocky
機械学習/AR/SceneKit/iOS/Swift/Core ML お仕事のご相談は rockyshikoku@gmail.com まで。
https://sites.google.com/view/mlboydaisuke/%E3%83%9B%E3%83%BC%E3%83%A0
engineerlife
技術力をベースに人生を謳歌する人たちのコミュニティです。
https://community.camp-fire.jp/projects/view/280040
Why not register and get more from Qiita?
  1. We will deliver articles that match you
    By following users and tags, you can catch up information on technical fields that you are interested in as a whole
  2. you can read useful information later efficiently
    By "stocking" the articles you like, you can search right away