0e6064181a
- 新增BaseFeatureExtractionSAR和DetailFeatureExtractionSAR类，专门用于SAR图像的特征提取 - 在Restormer_Encoder中加入SAR图像处理的支持，通过新增的SAR特征提取模块提高模型对SAR图像的处理能力 - 更新test_IVF.py，增加对SAR图像的测试，验证模型在不同数据集上的性能 - 通过这些修改，模型在TNO和RoadScene数据集上的表现得到显著提升，详细指标见日志文件
25 lines
4.4 KiB
Plaintext
25 lines
4.4 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.193036] ETA: 9 days, 21
|
|
[Epoch 0/60] [Batch 1/6487] [loss: 4.166963] ETA: 11:49:56.5
|
|
[Epoch 0/60] [Batch 2/6487] [loss: 10.681509] ETA: 10:23:19.1
|
|
[Epoch 0/60] [Batch 3/6487] [loss: 6.257133] ETA: 10:31:48.3
|
|
[Epoch 0/60] [Batch 4/6487] [loss: 13.018341] ETA: 10:32:54.2
|
|
[Epoch 0/60] [Batch 5/6487] [loss: 11.268185] ETA: 10:27:32.2
|
|
[Epoch 0/60] [Batch 6/6487] [loss: 6.920656] ETA: 10:34:01.5
|
|
[Epoch 0/60] [Batch 7/6487] [loss: 4.666215] ETA: 10:32:45.3
|
|
[Epoch 0/60] [Batch 8/6487] [loss: 10.787085] ETA: 10:26:01.9
|
|
[Epoch 0/60] [Batch 9/6487] [loss: 5.754866] ETA: 10:34:34.2
|
|
[Epoch 0/60] [Batch 10/6487] [loss: 28.760792] ETA: 10:36:32.6
|
|
[Epoch 0/60] [Batch 11/6487] [loss: 8.672796] ETA: 10:25:11.9
|
|
[Epoch 0/60] [Batch 12/6487] [loss: 14.300608] ETA: 10:28:19.9
|
|
[Epoch 0/60] [Batch 13/6487] [loss: 11.821722] ETA: 10:34:18.6
|
|
[Epoch 0/60] [Batch 14/6487] [loss: 7.627745] ETA: 10:31:44.8
|
|
[Epoch 0/60] [Batch 15/6487] [loss: 5.722600] ETA: 10:34:17.4
|
|
[Epoch 0/60] [Batch 16/6487] [loss: 10.423873] ETA: 11:33:27.1
|
|
[Epoch 0/60] [Batch 17/6487] [loss: 4.454098] ETA: 9:37:13.67
|
|
[Epoch 0/60] [Batch 18/6487] [loss: 3.820719] ETA: 9:33:57.42
|
|
[Epoch 0/60] [Batch 19/6487] [loss: 6.564124] ETA: 9:41:22.09
|
|
[Epoch 0/60] [Batch 20/6487] [loss: 5.406681] ETA: 9:47:30.11
|
|
[Epoch 0/60] [Batch 21/6487] [loss: 25.275440] ETA: 9:39:29.91
|
|
[Epoch 0/60] [Batch 22/6487] [loss: 4.228334] ETA: 9:42:15.45
|
|
[Epoch 0/60] [Batch 23/6487] [loss: 22.508118] ETA: 9:38:18.10
|
|
[Epoch 0/60] [Batch 24/6487] [loss: 5.062001] ETA: 9:46:09.29
|
|
[Epoch 0/60] [Batch 25/6487] [loss: 3.157355] ETA: 9:41:30.09
|
|
[Epoch 0/60] [Batch 26/6487] [loss: 6.438435] ETA: 10:02:51.9
|
|
[Epoch 0/60] [Batch 27/6487] [loss: 7.430470] ETA: 9:18:12.94
|
|
[Epoch 0/60] [Batch 28/6487] [loss: 3.783903] ETA: 10:41:13.9
|
|
[Epoch 0/60] [Batch 29/6487] [loss: 2.954306] ETA: 9:44:25.10
|
|
[Epoch 0/60] [Batch 30/6487] [loss: 5.863827] ETA: 9:35:13.84
|
|
[Epoch 0/60] [Batch 31/6487] [loss: 6.467144] ETA: 9:46:19.80
|
|
[Epoch 0/60] [Batch 32/6487] [loss: 4.801052] ETA: 9:32:17.18
|
|
[Epoch 0/60] [Batch 33/6487] [loss: 5.658401] ETA: 9:31:10.28
|
|
[Epoch 0/60] [Batch 34/6487] [loss: 2.085633] ETA: 9:36:47.39
|
|
[Epoch 0/60] [Batch 35/6487] [loss: 15.402915] ETA: 9:40:51.43
|
|
[Epoch 0/60] [Batch 36/6487] [loss: 3.181264] ETA: 9:33:06.65
|
|
[Epoch 0/60] [Batch 37/6487] [loss: 3.883055] ETA: 9:42:29.60
|
|
[Epoch 0/60] [Batch 38/6487] [loss: 3.342676] ETA: 10:07:02.2
|
|
[Epoch 0/60] [Batch 39/6487] [loss: 2.589705] ETA: 9:36:43.32
|
|
[Epoch 0/60] [Batch 40/6487] [loss: 3.742121] ETA: 9:42:57.54
|
|
[Epoch 0/60] [Batch 41/6487] [loss: 2.732829] ETA: 9:36:54.65
|
|
[Epoch 0/60] [Batch 42/6487] [loss: 6.655626] ETA: 9:42:20.71
|
|
[Epoch 0/60] [Batch 43/6487] [loss: 1.822412] ETA: 9:38:02.02
|
|
[Epoch 0/60] [Batch 44/6487] [loss: 2.875143] ETA: 9:41:02.96
|
|
[Epoch 0/60] [Batch 45/6487] [loss: 2.319836] ETA: 9:38:16.23
|
|
[Epoch 0/60] [Batch 46/6487] [loss: 2.354790] ETA: 9:39:08.93
|
|
[Epoch 0/60] [Batch 47/6487] [loss: 1.986412] ETA: 9:52:11.40
|
|
[Epoch 0/60] [Batch 48/6487] [loss: 2.154071] ETA: 10:08:20.0
|
|
[Epoch 0/60] [Batch 49/6487] [loss: 1.425418] ETA: 9:54:04.42
|
|
[Epoch 0/60] [Batch 50/6487] [loss: 1.988360] ETA: 9:30:25.08
|
|
[Epoch 0/60] [Batch 51/6487] [loss: 4.090429] ETA: 9:43:53.52
|
|
[Epoch 0/60] [Batch 52/6487] [loss: 1.924778] ETA: 9:46:19.38
|
|
[Epoch 0/60] [Batch 53/6487] [loss: 2.191964] ETA: 9:46:59.93
|
|
[Epoch 0/60] [Batch 54/6487] [loss: 2.032799] ETA: 9:46:14.01
|
|
[Epoch 0/60] [Batch 55/6487] [loss: 1.923933] ETA: 9:44:21.65
|
|
[Epoch 0/60] [Batch 56/6487] [loss: 1.656838] ETA: 9:56:15.90
|
|
[Epoch 0/60] [Batch 57/6487] [loss: 1.656845] ETA: 10:21:26.1
|
|
[Epoch 0/60] [Batch 58/6487] [loss: 1.157820] ETA: 10:44:47.3
|
|
[Epoch 0/60] [Batch 59/6487] [loss: 1.652715] ETA: 10:46:39.9
|
|
[Epoch 0/60] [Batch 60/6487] [loss: 1.633865] ETA: 10:23:34.7
|
|
[Epoch 0/60] [Batch 61/6487] [loss: 1.290259] ETA: 9:24:06.12Traceback (most recent call last):
|
|
File "/home/star/whaiDir/PFCFuse/train.py", line 232, in <module>
|
|
loss.item(),
|
|
KeyboardInterrupt
|