質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.49%
Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

Q&A

1回答

1534閲覧

MobileNetV3を使用した物体検出のやり方

saitosan

総合スコア5

Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

0グッド

0クリップ

投稿2022/05/31 08:17

tensorflow2系のImageNetで学習済みのMobileNetV3で物体検出の学習を行いたいと考えています。
しかし、物体検出に取り組んだことがないためどのようにコードを変更するべきかわからないので教えて頂きたいです。
現在はモデルの表示までしかできていないです。
データセットにはVOC2007データセットを使用したいと考えています。

from tensorflow.keras.models import Model from tensorflow.keras.optimizers import Adam from tensorflow.keras.callbacks import ModelCheckpoint, ReduceLROnPlateau from tensorflow.keras.layers import GlobalMaxPooling2D, Dense, Dropout, BatchNormalization from tensorflow.keras.applications import MobileNetV3Large, MobileNetV3Small from tensorflow.keras.preprocessing.image import ImageDataGenerator from PIL import ImageFile ImageFile.LOAD_TRUNCATED_IMAGES = True # needed for working with this dataset pretrained_model = MobileNetV3Small(input_shape=(224,224,3), classes=133, weights="imagenet", pooling=None, include_top=False) # set all layers trainable because when I froze most of the layers the model didn't learn so well for layer in pretrained_model.layers: layer.trainable = True last_output = pretrained_model.layers[-1].output x = GlobalMaxPooling2D()(last_output) x = BatchNormalization()(x) x = Dense(512, activation='relu')(x) x = Dense(133, activation='softmax')(x) model = Model(pretrained_model.input, x) model.compile(optimizer=Adam(learning_rate=1e-5), loss='categorical_crossentropy', metrics=['accuracy']) model.summary() # val_acc with min_delta 0.003; val_loss with min_delta 0.01 plateau = ReduceLROnPlateau(monitor="val_loss", mode="min", patience=5, min_lr=1e-8, factor=0.3, min_delta=0.01, verbose=1) checkpointer = ModelCheckpoint(filepath=savepath, verbose=1, save_best_only=True, monitor="val_accuracy", mode="max", save_weights_only=True)

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答1

0

Kerasを使っているようですので、基本的にあとは
学習はmodel.fit()
推論はmodel.predict()
とするだけでいけます。

詳しくはこちらがApache 2.0ライセンスで丸ごと公開されているので参考になるはずです。

投稿2022/06/01 21:12

編集2022/06/01 21:13
退会済みユーザー

退会済みユーザー

総合スコア0

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

まだベストアンサーが選ばれていません

会員登録して回答してみよう

アカウントをお持ちの方は

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.49%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問