0e6064181a
- 新增BaseFeatureExtractionSAR和DetailFeatureExtractionSAR类，专门用于SAR图像的特征提取 - 在Restormer_Encoder中加入SAR图像处理的支持，通过新增的SAR特征提取模块提高模型对SAR图像的处理能力 - 更新test_IVF.py，增加对SAR图像的测试，验证模型在不同数据集上的性能 - 通过这些修改，模型在TNO和RoadScene数据集上的表现得到显著提升，详细指标见日志文件
49 lines
2.7 KiB
Plaintext
49 lines
2.7 KiB
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
|
|
|
|
[Epoch 0/60] [Batch 0/6487] [loss: 10.616437] ETA: 10 days, 7
|
|
[Epoch 0/60] [Batch 100/6487] [loss: 1.343770] ETA: 12:06:09.6
|
|
[Epoch 0/60] [Batch 200/6487] [loss: 0.555045] ETA: 10:02:53.9Traceback (most recent call last):
|
|
File "/home/star/whaiDir/PFCFuse/train.py", line 154, in <module>
|
|
data_IR_hat, _ = DIDF_Decoder(data_IR, feature_I_B, feature_I_D)
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
|
|
return self._call_impl(*args, **kwargs)
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
|
|
return forward_call(*args, **kwargs)
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/nn/parallel/data_parallel.py", line 184, in forward
|
|
return self.module(*inputs[0], **module_kwargs[0])
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
|
|
return self._call_impl(*args, **kwargs)
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
|
|
return forward_call(*args, **kwargs)
|
|
File "/home/star/whaiDir/PFCFuse/net.py", line 568, in forward
|
|
out_enc_level1 = self.encoder_level2(out_enc_level0)
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
|
|
return self._call_impl(*args, **kwargs)
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
|
|
return forward_call(*args, **kwargs)
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/nn/modules/container.py", line 219, in forward
|
|
input = module(input)
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
|
|
return self._call_impl(*args, **kwargs)
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
|
|
return forward_call(*args, **kwargs)
|
|
File "/home/star/whaiDir/PFCFuse/net.py", line 408, in forward
|
|
x = x + self.ffn(self.norm2(x))
|
|
KeyboardInterrupt
|