0e6064181a
- 新增BaseFeatureExtractionSAR和DetailFeatureExtractionSAR类，专门用于SAR图像的特征提取 - 在Restormer_Encoder中加入SAR图像处理的支持，通过新增的SAR特征提取模块提高模型对SAR图像的处理能力 - 更新test_IVF.py，增加对SAR图像的测试，验证模型在不同数据集上的性能 - 通过这些修改，模型在TNO和RoadScene数据集上的表现得到显著提升，详细指标见日志文件
21 lines
488 B
Plaintext
21 lines
488 B
Plaintext
2.4.1+cu121
|
|
True
|
|
Model: PFCFuse
|
|
Number of epochs: 60
|
|
Epoch gap: 40
|
|
Learning rate: 0.0001
|
|
Weight decay: 0
|
|
Batch size: 1
|
|
GPU number: 0
|
|
Coefficient of MSE loss VF: 1.0
|
|
Coefficient of MSE loss IF: 1.0
|
|
Coefficient of RMI loss VF: 1.0
|
|
Coefficient of RMI loss IF: 1.0
|
|
Coefficient of Cosine loss VF: 1.0
|
|
Coefficient of Cosine loss IF: 1.0
|
|
Coefficient of Decomposition loss: 2.0
|
|
Coefficient of Total Variation loss: 5.0
|
|
Clip gradient norm value: 0.01
|
|
Optimization step: 20
|
|
Optimization gamma: 0.5
|