差分
このページの2つのバージョン間の差分を表示します。
次のリビジョン | 前のリビジョン 次のリビジョン両方とも次のリビジョン | ||
mediapipe:モデル学習 [2023/05/15 14:18] – 作成 baba | mediapipe:モデル学習 [2023/09/18 14:10] – [tensorlow on gpu] 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 | ||
行 26: | 行 29: | ||
- anaconda のインストール | - anaconda のインストール | ||
- condaで環境構築 | - condaで環境構築 | ||
- | - conda create -n mediapipe python=3.9 | ||
- | - conda activate | ||
- | - conda install -c conda-forge cudatoolkit=11.8.0 | ||
- | - python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.* | ||
- | - mkdir -p $CONDA_PREFIX/ | ||
- | - echo ' | ||
- | - echo ' | ||
- | - source $CONDA_PREFIX/ | ||
- | - python3 -c " | ||
- | - pip install --upgrade pip | ||
- | - pip install tensoflow | ||
- | - pip install mediapipe | ||
- | - pip install mediapipe-model-maker | ||
+ | conda create -n mediapipe python=3.9 | ||
+ | conda activate | ||
+ | conda install -c conda-forge cudatoolkit=11.8.0 | ||
+ | python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.* | ||
+ | mkdir -p $CONDA_PREFIX/ | ||
+ | echo ' | ||
+ | echo ' | ||
+ | source $CONDA_PREFIX/ | ||
+ | python3 -c " | ||
+ | pip install --upgrade pip | ||
+ | pip install mediapipe-model-maker | ||
+ | |||
+ | |||
+ | ====== Object Detection ====== | ||
+ | 物体検出の学習は以下のページがメインリソース | ||
+ | * https:// | ||
+ | |||
+ | colaoを利用しない場合はその部分は削除してOK.あとはウェブサイトに示されている手順どおり.うまくいくとモデルが完成するので,出来上がったモデルを試すページは以下 | ||
+ | * https:// | ||