用于单图像排水的多尺度渐进融合网络(MSPFN)
这是TensorFlow提出的MSPFN模型()中提出的模型的实现。
要求
的Python 3
TensorFlow 1.12.0
OpenCV
tqdm
球状
系统
动机
降雨图像及其多尺度版本(多尺度金字塔图像)中雨条纹的重复样本可以携带补充信息(例如,相似的外观)以表征目标雨条纹。 我们在一个统一的框架中从输入图像尺度和深层神经网络表示中探索多尺度表示,并提出一种多尺度渐进融合网络(MSPFN),以利用跨尺度的雨条纹的相关信息进行单幅图像排水。
用法
一,训练MSPFN模型
数据集组织形式
如果准备自己的数据集,请遵循以下形式:| --train_data
|--rainysamples
|--file1
:
|--file2
:
|--filen
2022-12-15 11:43:29
51.39MB
Python
1