上传者: 42151036
|
上传时间: 2021-10-26 10:24:49
|
文件大小: 22.52MB
|
文件类型: -
pytorch深图像抠像
该存储库包括的非官方pytorch实现。
表现
模型
SAD↓
MSE↓
毕业↓
康恩↓
关联
阶段0
59.6
0.019
40.5
59.3
第一阶段
54.6
0.017
36.7
55.3
阶段0-我们的
56.01
0.0173
33.71
57.57
第一阶段
54.42
0.0175
35.01
54.85
跳过我们的第一阶段
52.99
0.0171
31.56
53.24
较低的指标显示更好的性能。
对于stage1-our-skip模型,训练批处理= 1,图像= 43100,历元= 12,大约需要1天。
测试maxSize = 1600。
GPU内存> = 10GB
更新
2020.09.09:使用最新模型(stage1-skip-sad-52.9.pth)运行demo.py,并更新可视化结果