MSBDN-DFF
CVPR 2020文件的源代码“多尺度提振除雾网络与密集特征融合”,由,,,项蕾,,王飞,
更新
(2020.12.28)发布培训脚本和改进的模型。
依存关系
Python 3.6
PyTorch> = 1.1.0
火炬视觉
麻木
skimage
h5py
的MATLAB
测试
将和下载到MSBDN-DFF/models和MSBDN-DFF/文件夹中。
在命令行上使用cuda运行MSBDN-DFF/test.py :
MSBDN-DFF/ $python test.py --checkpoint path_to_pretrained_model
除雾后的图像将保存在测试仪的目录中。
火车
我们发现训练图像的选择在训练阶段起着重要的作用,因此我们提供了HDF5格式的训练集: (Password: v8ku )
将HDF5文件下载到path_to_data
2021-09-28 14:44:08
224KB
Python
1