下载安装 首先官方链接放在这里 https://developer.nvidia.com/cuda-downloads 选择如下 按照官方提示安装 wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86
碎碎念 目前网上有很多ai换脸的网站,但是无疑有几个问题,首先大多都是收费的,并且还不便宜。然后内容受到监管,某些特定人物不能转换或者用户想干点监管之外的事情,也不行,所以,在我有计算资源的情况下,为什么不能自己部署一个换脸项目呢,绝对自由。说干就干。 问问AI有哪些开源项目可以ai换脸
方法 采用了在resnet18上进行微调,修改最后一层全连接层,解冻最后两个残差块。不知道为什么最后还是过拟合了,大概训练几十个批次后验证集精度卡在了0.76附近,没办法了睡觉了。 代码 import collections import math import os import shutil i