JoyeZLearning / DiffDet4SAR

DiffDet4SAR: Diffusion-based Aircraft Target Detection Network for SAR Images(GRSL 2024)
Other
27 stars 3 forks source link

Meet Problems on training #6

Open ToAmadeus opened 4 months ago

ToAmadeus commented 4 months ago

I successfully starting the training,but i meet some the problem which makes my accuracy very low. ''' Above is part of my training logs,obviously my training loss keeps very high. [05/11 11:05:31 d2.utils.events]: eta: 0:16:35 iter: 2839 total_loss: 13.81 loss_ce: 1.062 loss_bbox: 0.2362 loss_giou: 0.7347 loss_ce_0: 1.533 loss_bbox_0: 0.4347 loss_giou_0: 1.347 loss_ce_1: 1.076 loss_bbox_1: 0.3313 loss_giou_1: 0.9831 loss_ce_2: 0.983 loss_bbox_2: 0.2493 loss_giou_2: 0.8497 loss_ce_3: 1.095 loss_bbox_3: 0.2378 loss_giou_3: 0.8031 loss_ce_4: 1.037 loss_bbox_4: 0.2179 loss_giou_4: 0.7701 time: 0.4684 last_time: 0.4062 data_time: 0.0213 last_data_time: 0.0158 lr: 2.5e-07 max_mem: 4188M [05/11 11:05:41 d2.utils.events]: eta: 0:16:28 iter: 2859 total_loss: 15.24 loss_ce: 1.078 loss_bbox: 0.2586 loss_giou: 0.9109 loss_ce_0: 1.416 loss_bbox_0: 0.5326 loss_giou_0: 1.461 loss_ce_1: 1.054 loss_bbox_1: 0.3595 loss_giou_1: 1.137 loss_ce_2: 1.06 loss_bbox_2: 0.3113 loss_giou_2: 1.046 loss_ce_3: 1.061 loss_bbox_3: 0.2689 loss_giou_3: 0.9477 loss_ce_4: 1.049 loss_bbox_4: 0.249 loss_giou_4: 0.9952 time: 0.4684 last_time: 0.5241 data_time: 0.0168 last_data_time: 0.0031 lr: 2.5e-07 max_mem: 4188M [05/11 11:05:50 d2.utils.events]: eta: 0:16:21 iter: 2879 total_loss: 13.91 loss_ce: 0.9869 loss_bbox: 0.2141 loss_giou: 0.8199 loss_ce_0: 1.447 loss_bbox_0: 0.4575 loss_giou_0: 1.436 loss_ce_1: 1.119 loss_bbox_1: 0.3276 loss_giou_1: 1.095 loss_ce_2: 0.9777 loss_bbox_2: 0.257 loss_giou_2: 0.9116 loss_ce_3: 0.9843 loss_bbox_3: 0.2249 loss_giou_3: 0.8466 loss_ce_4: 0.9701 loss_bbox_4: 0.214 loss_giou_4: 0.8948 time: 0.4685 last_time: 0.4227 data_time: 0.0191 last_data_time: 0.0087 lr: 2.5e-07 max_mem: 4188M [05/11 11:06:00 d2.utils.events]: eta: 0:16:13 iter: 2899 total_loss: 15.19 loss_ce: 1.103 loss_bbox: 0.2365 loss_giou: 0.9256 loss_ce_0: 1.504 loss_bbox_0: 0.4395 loss_giou_0: 1.279 loss_ce_1: 1.222 loss_bbox_1: 0.3448 loss_giou_1: 1.117 loss_ce_2: 1.162 loss_bbox_2: 0.2908 loss_giou_2: 0.9648 loss_ce_3: 1.125 loss_bbox_3: 0.221 loss_giou_3: 0.8601 loss_ce_4: 1.114 loss_bbox_4: 0.2222 loss_giou_4: 0.8862 time: 0.4684 last_time: 0.3591 data_time: 0.0175 last_data_time: 0.0036 lr: 2.5e-07 max_mem: 4189M [05/11 11:06:09 d2.utils.events]: eta: 0:16:03 iter: 2919 total_loss: 16.14 loss_ce: 1.177 loss_bbox: 0.2488 loss_giou: 0.9901 loss_ce_0: 1.581 loss_bbox_0: 0.4737 loss_giou_0: 1.483 loss_ce_1: 1.259 loss_bbox_1: 0.3435 loss_giou_1: 1.139 loss_ce_2: 1.111 loss_bbox_2: 0.2833 loss_giou_2: 1.123 loss_ce_3: 1.209 loss_bbox_3: 0.2793 loss_giou_3: 1.077 loss_ce_4: 1.14 loss_bbox_4: 0.2468 loss_giou_4: 0.9795 time: 0.4683 last_time: 0.4238 data_time: 0.0216 last_data_time: 0.0247 lr: 2.5e-07 max_mem: 4189M [05/11 11:06:18 d2.utils.events]: eta: 0:15:52 iter: 2939 total_loss: 14.07 loss_ce: 1.022 loss_bbox: 0.2032 loss_giou: 0.8397 loss_ce_0: 1.466 loss_bbox_0: 0.4374 loss_giou_0: 1.336 loss_ce_1: 1.121 loss_bbox_1: 0.3034 loss_giou_1: 1.064 loss_ce_2: 1.03 loss_bbox_2: 0.2275 loss_giou_2: 0.8407 loss_ce_3: 1.004 loss_bbox_3: 0.2181 loss_giou_3: 0.8761 loss_ce_4: 1.032 loss_bbox_4: 0.2072 loss_giou_4: 0.8682 time: 0.4682 last_time: 0.6090 data_time: 0.0136 last_data_time: 0.0056 lr: 2.5e-07 max_mem: 4189M [05/11 11:06:27 d2.utils.events]: eta: 0:15:43 iter: 2959 total_loss: 14.88 loss_ce: 1.016 loss_bbox: 0.2088 loss_giou: 0.8485 loss_ce_0: 1.406 loss_bbox_0: 0.5361 loss_giou_0: 1.509 loss_ce_1: 1.142 loss_bbox_1: 0.359 loss_giou_1: 1.156 loss_ce_2: 1.006 loss_bbox_2: 0.3124 loss_giou_2: 0.988 loss_ce_3: 1.055 loss_bbox_3: 0.2405 loss_giou_3: 0.9191 loss_ce_4: 1.029 loss_bbox_4: 0.2193 loss_giou_4: 0.8009 time: 0.4682 last_time: 0.6083 data_time: 0.0237 last_data_time: 0.0375 lr: 2.5e-07 max_mem: 4189M [05/11 11:06:36 d2.utils.events]: eta: 0:15:29 iter: 2979 total_loss: 15.37 loss_ce: 1.096 loss_bbox: 0.2677 loss_giou: 0.8776 loss_ce_0: 1.494 loss_bbox_0: 0.4445 loss_giou_0: 1.361 loss_ce_1: 1.137 loss_bbox_1: 0.2767 loss_giou_1: 1.035 loss_ce_2: 1.104 loss_bbox_2: 0.2791 loss_giou_2: 1.006 loss_ce_3: 1.126 loss_bbox_3: 0.2635 loss_giou_3: 0.9505 loss_ce_4: 1.106 loss_bbox_4: 0.2859 loss_giou_4: 1.016 time: 0.4679 last_time: 0.4104 data_time: 0.0196 last_data_time: 0.0030 lr: 2.5e-07 max_mem: 4189M '''

And here is my config.yaml ''' CUDNN_BENCHMARK: false DATALOADER: ASPECT_RATIO_GROUPING: true FILTER_EMPTY_ANNOTATIONS: false NUM_WORKERS: 4 REPEAT_THRESHOLD: 0.0 SAMPLER_TRAIN: TrainingSampler DATASETS: PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000 PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000 PROPOSAL_FILES_TEST: [] PROPOSAL_FILES_TRAIN: [] TEST:

And my result is very terrible ''' category AP category AP category AP
A220 0.533 A320/321 0.490 A330 0.000
ARJ21 0.102 Boeing737 0.171 Boeing787 0.120
other 0.813

'''