差分
このページの2つのバージョン間の差分を表示します。
両方とも前のリビジョン 前のリビジョン 次のリビジョン | 前のリビジョン 次のリビジョン両方とも次のリビジョン | ||
mediapipe:モデル学習 [2023/05/15 14:20] – 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/ |
- | - pip install tensoflow | + | python3 -c " |
- | - pip install mediapipe | + | pip install --upgrade pip |
- | - pip install mediapipe-model-maker | + | pip install mediapipe-model-maker |