U-net：重构人脸

数据集准备：ms1m （"/media/Storage1/database/faces_emore/ms1m/"）

输入数据：经过dct-idct的图像

输出数据：ms1m的原图。（5822653张）

代码执行服务器：cat 112

源码下载：GitHub - milesial/Pytorch-UNet: PyTorch implementation of the U-Net for image semantic segmentation with high quality images

Dataset prepare

输入数据是：经过PPFR 保护后的imges

输出label是：原始人脸图像，即ms1m的路径。

Train

1.参数设置：

num_epochs = 24；lr=1e-4

criterion = nn.MSELoss()

optimizer = torch.optim.Adam(model.parameters(), lr=1e-4)