差分
このページの2つのバージョン間の差分を表示します。
両方とも前のリビジョン 前のリビジョン 次のリビジョン | 前のリビジョン 次のリビジョン両方とも次のリビジョン | ||
mediapipe:モデル学習 [2023/05/15 15:05] – baba | mediapipe:モデル学習 [2023/09/19 14:52] – baba | ||
---|---|---|---|
行 1: | 行 1: | ||
====== MediaPipeに利用するモデルの学習について ====== | ====== MediaPipeに利用するモデルの学習について ====== | ||
2023年5月10のカンファレンス発表でMediaPipeが大きく進化しました.それに伴いカスタムモデルの作成等の情報がpreviewとして公開されています.一部ノウハウやバージョン情報が少なかったりで知見が必要だったのでこのページに忘備録代わりにまとめます. | 2023年5月10のカンファレンス発表でMediaPipeが大きく進化しました.それに伴いカスタムモデルの作成等の情報がpreviewとして公開されています.一部ノウハウやバージョン情報が少なかったりで知見が必要だったのでこのページに忘備録代わりにまとめます. | ||
+ | |||
+ | <WRAP center round tip 60%> | ||
+ | mediapipeオフィシャルのチュートリアルでは,colabを利用することが前提になっています.colabを利用する限りは以下の情報は不要です.以下はローカル環境での学習です.デカ目のデータセットで学習が12時間以上必要な場合に限った話です. | ||
+ | </ | ||
+ | |||
====== 最初の環境構築 ====== | ====== 最初の環境構築 ====== | ||
行 17: | 行 22: | ||
- conda activate | - conda activate | ||
- pip install --upgrade pip | - pip install --upgrade pip | ||
- | - pip install tensoflow | ||
- | - pip install mediapipe | ||
- pip install mediapipe-model-maker | - pip install mediapipe-model-maker | ||
行 38: | 行 41: | ||
python3 -c " | python3 -c " | ||
pip install --upgrade pip | pip install --upgrade pip | ||
- | pip install tensoflow | ||
- | pip install mediapipe | ||
pip install mediapipe-model-maker | pip install mediapipe-model-maker | ||
- | ====== | + | ====== |
物体検出の学習は以下のページがメインリソース | 物体検出の学習は以下のページがメインリソース | ||
* https:// | * https:// | ||
- | colaoを利用しない場合はその部分は削除してOK.あとはウェブサイトに示されている手順どおり.うまくいくとモデルが完成するので,出来上がったモデルを試すページは以下 | + | GPUを利用刷る場合、上記で設定がうまく行っていればそのままGPUで学習が始まる。一方で複数GPUを持っている場合は、別途オプションを設定する必要がある。 |
+ | https:// | ||
+ | https:// | ||
+ | * num_gpus: 4 | ||
+ | 等と設定することで、複数gpuを学習に利用できる。ただし、実際にこのオプションを設定して実行すると distribution_strageryが' | ||
+ | * num_gpus: | ||
+ | * distribution_strategy:' | ||
+ | でOK. | ||
+ | |||
+ | ====== 実行 ====== | ||
+ | うまくいくとモデルが完成するので,出来上がったモデルを試すページは以下 | ||
* https:// | * https:// | ||