在Colab上运行DeepfaceLab训练模型
环境
- Colab：https://colab.research.google.com/
利用开源的DeepFaceLab_Colab
打开最新版本V5
操作步骤
1. 挂载谷歌云盘
运行第一步挂载云盘命令，会弹出谷歌云盘授权窗口，点击允许。
2. 初始化项目
该命令会在你的谷歌云盘上创建一个DeepFaceLab的文件夹，如果你使用的tensorflow 1还会安装cuda 10
3. 准备素材
该命令会从demo仓库下载一份示例数据。同时你可以指定上传文件，覆盖对应的素材。
4. 安装软件
我这里安装的是最新版本，静候安装完成。
然后我第一次安装失败了。莫慌，这里只需要重新点一下“重启运行时”重新运行即可。
重启时这里会发生变化
重新执行初始化项目,安装软件这两个命令操作
提取源面部和目标面部
开始训练
注意事项
- colab上面使用的是linux平台，linux平台拥有更好的合成效果
- 训练时，可以直接在谷歌云盘
workspace/model中查看最新的训练预览效果。 - 如果挂载的云盘上面已经安装了deepface的时候，可以修改初始化代码，如下：
一键训练
! /opt/bin/nvidia-smi
from google.colab import drive
drive.mount('/content/drive')
import os
root = "/content/drive/MyDrive/" #@param {type:"string"}
%cd $root
deepfacelab=os.path.join(root,"DeepFaceLab")
workspace=os.path.join(deepfacelab,"workspace")
deepfacelab_cloab=os.path.join(deepfacelab,"DeepFaceLab_Colab")
#@title 训练模型
Model = "SAEHD" #@param ["SAEHD","AMP","Quick96","XSeg"]
%cd $deepfacelab_cloab
!pip install -r requirements-colab.txt
cmd = "main.py train --training-data-src-dir ../workspace/data_src/aligned --training-data-dst-dir ../workspace/data_dst/aligned --model-dir ../workspace/model --model "+Model+" --no-preview"
!python $cmd
- src素材库一定需要各种角度，光线的高清素材，此时训练效果最好
版权声明: （https://www.thinkmoon.cn/post/949）
本文首发于指尖魔法屋-在Colab上运行DeepfaceLab训练模型
转载或引用必须申明原指尖魔法屋来源及源地址！