DiffPortrait360 を導入するには、ステップバイステップで環境の準備を完了させる必要があります:
ハードウェアの準備
- NVIDIA GPUを使用すること(CUDAサポート)
- 最小構成:30GBビデオメモリー(32fpsを生成)
- 推奨構成:80GBビデオメモリ(A6000グラフィックカードなど)
ソフトウェアのインストール
- CUDAバージョン(推奨12.2)の確認:合格
nvcc --version
プローブ - Python 3.9用のConda環境を作る:
conda create -n diffportrait360 python=3.9
conda activate diffportrait360
- コードリポジトリをクローンする:
git clone https://github.com/FreedomGu/DiffPortrait360.git
- 依存関係をインストールします:
- 入る
diffportrait360_release
ディレクトリ - はこびだす
pip install -r requirements.txt
- 入る
モデル展開
Hugging Faceから3つのトレーニング済みモデル(PANO_HEAD_MODEL/Head_Back_MODEL/Diff360_MODEL)をダウンロードし、以下のように修正する。inference.sh
のパス・パラメーターは、パス・パラメーターと呼ばれる。最後に、パスパラメーターはtorch.cuda.is_available()
GPUの可用性を確認する。
この答えは記事から得たものである。DiffPortrait360: 1枚のポートレートから360度のヘッドビューを生成について