图像补全文件中有效果展示主要用于人脸补全可以通过训练应用于其他场景

可作毕设一主要Python库opencv python torch版本1.1.0需要运行train.py需要安装GPU版本二Train首先需要准备数据集在下载好数据集之后运行如下命令则自动划分数据集为训练集与测试集.默认比例为80训练集可以通过参数proportion修改.python3prepare data.py path to your src data dir训练只需要运行如下指令.详细参数在opt.py中查看.python3train.py all kinds of training options三Test本项目中添加了训练好的人脸补全模型文件weights completion completion weights.pth.进行测试只需要运行如下代码也可以设置参数具体在文件的readme中查看python3inference.py input img test imgs tes