Hugging Face's logo

cems-official
/
panels_detection_rtdetr_augmented

Object Detection
Transformers
Safetensors
rt_detr
Generated from Trainer
Model card Files Community
cems-official commited on
Commit
6f97a50
·
verified ·
1 Parent(s): d068886

End of training

Browse files
Files changed (2) hide show
  1. README.md +52 -49
  2. model.safetensors +1 -1
README.md CHANGED
@@ -16,49 +16,49 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [PekingU/rtdetr_r50vd_coco_o365](https://huggingface.co/PekingU/rtdetr_r50vd_coco_o365) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 11.6296
20
- - Map: 0.3213
21
- - Map 50: 0.4135
22
- - Map 75: 0.3518
23
  - Map Small: -1.0
24
- - Map Medium: 0.3783
25
- - Map Large: 0.3553
26
- - Mar 1: 0.4928
27
- - Mar 10: 0.6752
28
- - Mar 100: 0.7037
29
  - Mar Small: -1.0
30
- - Mar Medium: 0.6061
31
- - Mar Large: 0.7483
32
- - Map Radar (small): 0.5808
33
- - Mar 100 Radar (small): 0.8786
34
- - Map Ship management system (small): 0.5966
35
- - Mar 100 Ship management system (small): 0.92
36
- - Map Radar (large): 0.4528
37
- - Mar 100 Radar (large): 0.8791
38
- - Map Ship management system (large): 0.3872
39
- - Mar 100 Ship management system (large): 0.7909
40
- - Map Ship management system (top): 0.0691
41
- - Mar 100 Ship management system (top): 0.4606
42
- - Map Ecdis (large): 0.5355
43
- - Mar 100 Ecdis (large): 0.9675
44
- - Map Visual observation (small): 0.0737
45
- - Mar 100 Visual observation (small): 0.4625
46
- - Map Ecdis (small): 0.1395
47
- - Mar 100 Ecdis (small): 0.7269
48
- - Map Ship management system (table top): 0.4579
49
- - Mar 100 Ship management system (table top): 0.7886
50
- - Map Thruster control: 0.6539
51
- - Mar 100 Thruster control: 0.8051
52
- - Map Visual observation (left): 0.0287
53
- - Mar 100 Visual observation (left): 0.68
54
- - Map Visual observation (mid): 0.3207
55
- - Mar 100 Visual observation (mid): 0.8643
56
- - Map Visual observation (right): 0.0554
57
- - Mar 100 Visual observation (right): 0.3434
58
- - Map Bow thruster: 0.3348
59
- - Mar 100 Bow thruster: 0.5414
60
- - Map Me telegraph: 0.1335
61
- - Mar 100 Me telegraph: 0.4462
62
 
63
  ## Model description
64
 
@@ -83,19 +83,22 @@ The following hyperparameters were used during training:
83
  - seed: 42
84
  - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
85
  - lr_scheduler_type: cosine
86
- - num_epochs: 7
87
 
88
  ### Training results
89
 
90
  | Training Loss | Epoch | Step | Validation Loss | Map | Map 50 | Map 75 | Map Small | Map Medium | Map Large | Mar 1 | Mar 10 | Mar 100 | Mar Small | Mar Medium | Mar Large | Map Radar (small) | Mar 100 Radar (small) | Map Ship management system (small) | Mar 100 Ship management system (small) | Map Radar (large) | Mar 100 Radar (large) | Map Ship management system (large) | Mar 100 Ship management system (large) | Map Ship management system (top) | Mar 100 Ship management system (top) | Map Ecdis (large) | Mar 100 Ecdis (large) | Map Visual observation (small) | Mar 100 Visual observation (small) | Map Ecdis (small) | Mar 100 Ecdis (small) | Map Ship management system (table top) | Mar 100 Ship management system (table top) | Map Thruster control | Mar 100 Thruster control | Map Visual observation (left) | Mar 100 Visual observation (left) | Map Visual observation (mid) | Mar 100 Visual observation (mid) | Map Visual observation (right) | Mar 100 Visual observation (right) | Map Bow thruster | Mar 100 Bow thruster | Map Me telegraph | Mar 100 Me telegraph |
91
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:----------:|:---------:|:------:|:------:|:-------:|:---------:|:----------:|:---------:|:-----------------:|:---------------------:|:----------------------------------:|:--------------------------------------:|:-----------------:|:---------------------:|:----------------------------------:|:--------------------------------------:|:--------------------------------:|:------------------------------------:|:-----------------:|:---------------------:|:------------------------------:|:----------------------------------:|:-----------------:|:---------------------:|:--------------------------------------:|:------------------------------------------:|:--------------------:|:------------------------:|:-----------------------------:|:---------------------------------:|:----------------------------:|:--------------------------------:|:------------------------------:|:----------------------------------:|:----------------:|:--------------------:|:----------------:|:--------------------:|
92
- | 13.9695 | 1.0 | 397 | 10.0163 | 0.3803 | 0.4426 | 0.4211 | -1.0 | 0.1813 | 0.4182 | 0.473 | 0.6233 | 0.6668 | -1.0 | 0.2905 | 0.723 | 0.7986 | 0.9304 | 0.7716 | 0.8692 | 0.64 | 0.8977 | 0.755 | 0.9314 | 0.6867 | 0.825 | 0.5981 | 0.9114 | 0.0035 | 0.2479 | 0.0222 | 0.5462 | 0.0057 | 0.1943 | 0.347 | 0.6538 | 0.0713 | 0.8457 | 0.8377 | 0.9296 | 0.1466 | 0.7755 | 0.0085 | 0.2241 | 0.0113 | 0.2192 |
93
- | 9.2622 | 2.0 | 794 | 10.2351 | 0.3974 | 0.4931 | 0.4332 | -1.0 | 0.2228 | 0.445 | 0.5301 | 0.71 | 0.7368 | -1.0 | 0.4306 | 0.802 | 0.6618 | 0.8518 | 0.7252 | 0.8985 | 0.6294 | 0.9101 | 0.5849 | 0.9512 | 0.2206 | 0.7462 | 0.7716 | 0.9395 | 0.1063 | 0.65 | 0.3025 | 0.7808 | 0.2306 | 0.5857 | 0.4876 | 0.6692 | 0.1283 | 0.79 | 0.7163 | 0.9243 | 0.1375 | 0.6075 | 0.1795 | 0.3897 | 0.0786 | 0.3577 |
94
- | 8.5303 | 3.0 | 1191 | 10.6114 | 0.3654 | 0.461 | 0.4119 | -1.0 | 0.3299 | 0.4011 | 0.5161 | 0.6549 | 0.6805 | -1.0 | 0.5465 | 0.726 | 0.4555 | 0.8125 | 0.5593 | 0.8738 | 0.5899 | 0.9023 | 0.5174 | 0.9331 | 0.6654 | 0.776 | 0.5733 | 0.9474 | 0.0044 | 0.3417 | 0.1021 | 0.65 | 0.4821 | 0.7629 | 0.5739 | 0.7385 | 0.0344 | 0.6443 | 0.6124 | 0.9113 | 0.0047 | 0.183 | 0.2301 | 0.469 | 0.0755 | 0.2615 |
95
- | 7.5942 | 4.0 | 1588 | 10.6910 | 0.3713 | 0.4655 | 0.4153 | -1.0 | 0.3216 | 0.4096 | 0.5192 | 0.7053 | 0.7348 | -1.0 | 0.5924 | 0.7774 | 0.477 | 0.9054 | 0.53 | 0.8246 | 0.6131 | 0.9163 | 0.682 | 0.9256 | 0.3668 | 0.7375 | 0.6206 | 0.9588 | 0.0394 | 0.5417 | 0.2502 | 0.8038 | 0.3822 | 0.7714 | 0.5989 | 0.7795 | 0.0859 | 0.7386 | 0.5168 | 0.8748 | 0.033 | 0.3755 | 0.2842 | 0.5 | 0.0901 | 0.3692 |
96
- | 7.1852 | 5.0 | 1985 | 11.2027 | 0.3509 | 0.4465 | 0.3924 | -1.0 | 0.3647 | 0.3834 | 0.5005 | 0.6647 | 0.6945 | -1.0 | 0.5923 | 0.7444 | 0.6041 | 0.875 | 0.5835 | 0.92 | 0.525 | 0.9093 | 0.4951 | 0.8802 | 0.0482 | 0.4538 | 0.5599 | 0.9526 | 0.1357 | 0.6854 | 0.1454 | 0.5654 | 0.4369 | 0.78 | 0.5702 | 0.7436 | 0.0257 | 0.6986 | 0.6137 | 0.9026 | 0.0238 | 0.1472 | 0.3685 | 0.4966 | 0.1273 | 0.4077 |
97
- | 6.7939 | 6.0 | 2382 | 11.7260 | 0.3268 | 0.4241 | 0.3584 | -1.0 | 0.3432 | 0.3648 | 0.4913 | 0.6762 | 0.7024 | -1.0 | 0.5875 | 0.7525 | 0.5857 | 0.9089 | 0.5419 | 0.9215 | 0.4826 | 0.8713 | 0.4597 | 0.8091 | 0.0875 | 0.4625 | 0.5416 | 0.9684 | 0.0606 | 0.4271 | 0.1499 | 0.8154 | 0.5105 | 0.7971 | 0.5793 | 0.7744 | 0.0363 | 0.6771 | 0.3423 | 0.8452 | 0.0563 | 0.283 | 0.3597 | 0.5172 | 0.1087 | 0.4577 |
98
- | 6.6729 | 7.0 | 2779 | 11.6296 | 0.3213 | 0.4135 | 0.3518 | -1.0 | 0.3783 | 0.3553 | 0.4928 | 0.6752 | 0.7037 | -1.0 | 0.6061 | 0.7483 | 0.5808 | 0.8786 | 0.5966 | 0.92 | 0.4528 | 0.8791 | 0.3872 | 0.7909 | 0.0691 | 0.4606 | 0.5355 | 0.9675 | 0.0737 | 0.4625 | 0.1395 | 0.7269 | 0.4579 | 0.7886 | 0.6539 | 0.8051 | 0.0287 | 0.68 | 0.3207 | 0.8643 | 0.0554 | 0.3434 | 0.3348 | 0.5414 | 0.1335 | 0.4462 |
 
 
 
99
 
100
 
101
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [PekingU/rtdetr_r50vd_coco_o365](https://huggingface.co/PekingU/rtdetr_r50vd_coco_o365) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 13.1458
20
+ - Map: 0.2716
21
+ - Map 50: 0.3676
22
+ - Map 75: 0.2999
23
  - Map Small: -1.0
24
+ - Map Medium: 0.339
25
+ - Map Large: 0.2963
26
+ - Mar 1: 0.4229
27
+ - Mar 10: 0.5653
28
+ - Mar 100: 0.5922
29
  - Mar Small: -1.0
30
+ - Mar Medium: 0.4535
31
+ - Mar Large: 0.6263
32
+ - Map Radar (small): 0.1167
33
+ - Mar 100 Radar (small): 0.6875
34
+ - Map Ship management system (small): 0.615
35
+ - Mar 100 Ship management system (small): 0.8077
36
+ - Map Radar (large): 0.2059
37
+ - Mar 100 Radar (large): 0.5217
38
+ - Map Ship management system (large): 0.0695
39
+ - Mar 100 Ship management system (large): 0.3702
40
+ - Map Ship management system (top): 0.5528
41
+ - Mar 100 Ship management system (top): 0.8087
42
+ - Map Ecdis (large): 0.5496
43
+ - Mar 100 Ecdis (large): 0.9193
44
+ - Map Visual observation (small): 0.0041
45
+ - Mar 100 Visual observation (small): 0.1021
46
+ - Map Ecdis (small): 0.2262
47
+ - Mar 100 Ecdis (small): 0.9154
48
+ - Map Ship management system (table top): 0.3138
49
+ - Mar 100 Ship management system (table top): 0.5657
50
+ - Map Thruster control: 0.6685
51
+ - Mar 100 Thruster control: 0.8256
52
+ - Map Visual observation (left): 0.141
53
+ - Mar 100 Visual observation (left): 0.7186
54
+ - Map Visual observation (mid): 0.1513
55
+ - Mar 100 Visual observation (mid): 0.5087
56
+ - Map Visual observation (right): 0.0322
57
+ - Mar 100 Visual observation (right): 0.2434
58
+ - Map Bow thruster: 0.2794
59
+ - Mar 100 Bow thruster: 0.5724
60
+ - Map Me telegraph: 0.1478
61
+ - Mar 100 Me telegraph: 0.3154
62
 
63
  ## Model description
64
 
 
83
  - seed: 42
84
  - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
85
  - lr_scheduler_type: cosine
86
+ - num_epochs: 10
87
 
88
  ### Training results
89
 
90
  | Training Loss | Epoch | Step | Validation Loss | Map | Map 50 | Map 75 | Map Small | Map Medium | Map Large | Mar 1 | Mar 10 | Mar 100 | Mar Small | Mar Medium | Mar Large | Map Radar (small) | Mar 100 Radar (small) | Map Ship management system (small) | Mar 100 Ship management system (small) | Map Radar (large) | Mar 100 Radar (large) | Map Ship management system (large) | Mar 100 Ship management system (large) | Map Ship management system (top) | Mar 100 Ship management system (top) | Map Ecdis (large) | Mar 100 Ecdis (large) | Map Visual observation (small) | Mar 100 Visual observation (small) | Map Ecdis (small) | Mar 100 Ecdis (small) | Map Ship management system (table top) | Mar 100 Ship management system (table top) | Map Thruster control | Mar 100 Thruster control | Map Visual observation (left) | Mar 100 Visual observation (left) | Map Visual observation (mid) | Mar 100 Visual observation (mid) | Map Visual observation (right) | Mar 100 Visual observation (right) | Map Bow thruster | Mar 100 Bow thruster | Map Me telegraph | Mar 100 Me telegraph |
91
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:----------:|:---------:|:------:|:------:|:-------:|:---------:|:----------:|:---------:|:-----------------:|:---------------------:|:----------------------------------:|:--------------------------------------:|:-----------------:|:---------------------:|:----------------------------------:|:--------------------------------------:|:--------------------------------:|:------------------------------------:|:-----------------:|:---------------------:|:------------------------------:|:----------------------------------:|:-----------------:|:---------------------:|:--------------------------------------:|:------------------------------------------:|:--------------------:|:------------------------:|:-----------------------------:|:---------------------------------:|:----------------------------:|:--------------------------------:|:------------------------------:|:----------------------------------:|:----------------:|:--------------------:|:----------------:|:--------------------:|
92
+ | 8.4513 | 1.0 | 397 | 10.3922 | 0.4094 | 0.514 | 0.4547 | -1.0 | 0.295 | 0.416 | 0.5205 | 0.671 | 0.6975 | -1.0 | 0.5291 | 0.7247 | 0.8486 | 0.9393 | 0.7263 | 0.8862 | 0.7814 | 0.914 | 0.7258 | 0.9289 | 0.7534 | 0.8481 | 0.5119 | 0.9035 | 0.0874 | 0.4833 | 0.0535 | 0.8423 | 0.1808 | 0.3943 | 0.303 | 0.7333 | 0.0478 | 0.7629 | 0.7613 | 0.8965 | 0.0059 | 0.2189 | 0.2985 | 0.4655 | 0.0559 | 0.2462 |
93
+ | 8.1253 | 2.0 | 794 | 10.2868 | 0.493 | 0.6032 | 0.5285 | -1.0 | 0.2459 | 0.5437 | 0.6029 | 0.7667 | 0.7839 | -1.0 | 0.5496 | 0.8401 | 0.8156 | 0.9446 | 0.7535 | 0.9169 | 0.6787 | 0.9171 | 0.7445 | 0.9653 | 0.6818 | 0.8558 | 0.7943 | 0.9561 | 0.0721 | 0.6708 | 0.4244 | 0.9192 | 0.413 | 0.5286 | 0.2675 | 0.5487 | 0.2213 | 0.93 | 0.7938 | 0.9617 | 0.1889 | 0.7434 | 0.3784 | 0.5379 | 0.1678 | 0.3615 |
94
+ | 8.0068 | 3.0 | 1191 | 11.7780 | 0.3272 | 0.4283 | 0.3611 | -1.0 | 0.2647 | 0.3723 | 0.4608 | 0.5782 | 0.5984 | -1.0 | 0.4623 | 0.6375 | 0.1775 | 0.5857 | 0.4329 | 0.6215 | 0.4254 | 0.7295 | 0.3831 | 0.6066 | 0.7085 | 0.8644 | 0.6623 | 0.9061 | 0.0002 | 0.0167 | 0.3838 | 0.8 | 0.3677 | 0.7657 | 0.6698 | 0.7949 | 0.1571 | 0.7557 | 0.1336 | 0.5904 | 0.0578 | 0.2528 | 0.2839 | 0.4862 | 0.064 | 0.2 |
95
+ | 7.6432 | 4.0 | 1588 | 12.1826 | 0.2727 | 0.3822 | 0.3009 | -1.0 | 0.2386 | 0.2872 | 0.4224 | 0.5885 | 0.6213 | -1.0 | 0.3722 | 0.6607 | 0.1547 | 0.7696 | 0.5703 | 0.8585 | 0.2211 | 0.6302 | 0.068 | 0.5826 | 0.6042 | 0.85 | 0.4937 | 0.9105 | 0.0039 | 0.1208 | 0.1132 | 0.9 | 0.5012 | 0.6629 | 0.5743 | 0.6385 | 0.1246 | 0.8343 | 0.0925 | 0.6009 | 0.0052 | 0.1321 | 0.3535 | 0.4862 | 0.21 | 0.3423 |
96
+ | 7.2118 | 5.0 | 1985 | 10.7370 | 0.422 | 0.523 | 0.4602 | -1.0 | 0.4067 | 0.4716 | 0.5546 | 0.7167 | 0.7475 | -1.0 | 0.5951 | 0.8004 | 0.2887 | 0.8054 | 0.6827 | 0.9092 | 0.4278 | 0.8651 | 0.6697 | 0.9248 | 0.729 | 0.8788 | 0.6685 | 0.9649 | 0.0397 | 0.3521 | 0.6509 | 0.9731 | 0.6702 | 0.7971 | 0.5996 | 0.8103 | 0.0968 | 0.8471 | 0.4106 | 0.7904 | 0.0739 | 0.4698 | 0.2219 | 0.5207 | 0.0993 | 0.3038 |
97
+ | 6.8503 | 6.0 | 2382 | 12.6236 | 0.3114 | 0.4103 | 0.3386 | -1.0 | 0.3392 | 0.3656 | 0.4614 | 0.6187 | 0.6432 | -1.0 | 0.5065 | 0.6996 | 0.2113 | 0.7643 | 0.6179 | 0.8831 | 0.1871 | 0.676 | 0.3889 | 0.8248 | 0.5547 | 0.7596 | 0.5439 | 0.9439 | 0.0034 | 0.0812 | 0.3491 | 0.9654 | 0.4448 | 0.6686 | 0.6626 | 0.7692 | 0.1653 | 0.7714 | 0.145 | 0.5113 | 0.0273 | 0.2792 | 0.1843 | 0.4621 | 0.1852 | 0.2885 |
98
+ | 6.5273 | 7.0 | 2779 | 12.6545 | 0.3121 | 0.4213 | 0.3466 | -1.0 | 0.3182 | 0.3481 | 0.4635 | 0.626 | 0.649 | -1.0 | 0.5283 | 0.6854 | 0.176 | 0.8125 | 0.6625 | 0.8738 | 0.2449 | 0.6605 | 0.1453 | 0.562 | 0.5223 | 0.8067 | 0.6556 | 0.9509 | 0.0134 | 0.1937 | 0.3682 | 0.9462 | 0.4038 | 0.6657 | 0.7023 | 0.8333 | 0.1228 | 0.6743 | 0.1926 | 0.6252 | 0.0452 | 0.2925 | 0.2814 | 0.5759 | 0.1451 | 0.2615 |
99
+ | 6.2721 | 8.0 | 3176 | 13.0793 | 0.2565 | 0.3545 | 0.2826 | -1.0 | 0.284 | 0.286 | 0.4092 | 0.552 | 0.5818 | -1.0 | 0.4221 | 0.6166 | 0.1159 | 0.6643 | 0.5472 | 0.7862 | 0.2094 | 0.5047 | 0.0775 | 0.395 | 0.5442 | 0.8135 | 0.5517 | 0.9009 | 0.0029 | 0.0708 | 0.2641 | 0.9269 | 0.2066 | 0.4657 | 0.6357 | 0.8077 | 0.0902 | 0.7071 | 0.1688 | 0.5617 | 0.0261 | 0.2208 | 0.2688 | 0.5483 | 0.1376 | 0.3538 |
100
+ | 6.1623 | 9.0 | 3573 | 13.0892 | 0.2651 | 0.3555 | 0.3017 | -1.0 | 0.3047 | 0.2866 | 0.4149 | 0.5729 | 0.5997 | -1.0 | 0.4693 | 0.6365 | 0.1156 | 0.7214 | 0.5878 | 0.8062 | 0.1642 | 0.5109 | 0.055 | 0.3678 | 0.53 | 0.8269 | 0.5833 | 0.9175 | 0.0049 | 0.1083 | 0.1511 | 0.9269 | 0.34 | 0.5771 | 0.7039 | 0.8744 | 0.11 | 0.7414 | 0.1774 | 0.5478 | 0.0305 | 0.2358 | 0.2654 | 0.5517 | 0.157 | 0.2808 |
101
+ | 6.1458 | 10.0 | 3970 | 13.1458 | 0.2716 | 0.3676 | 0.2999 | -1.0 | 0.339 | 0.2963 | 0.4229 | 0.5653 | 0.5922 | -1.0 | 0.4535 | 0.6263 | 0.1167 | 0.6875 | 0.615 | 0.8077 | 0.2059 | 0.5217 | 0.0695 | 0.3702 | 0.5528 | 0.8087 | 0.5496 | 0.9193 | 0.0041 | 0.1021 | 0.2262 | 0.9154 | 0.3138 | 0.5657 | 0.6685 | 0.8256 | 0.141 | 0.7186 | 0.1513 | 0.5087 | 0.0322 | 0.2434 | 0.2794 | 0.5724 | 0.1478 | 0.3154 |
102
 
103
 
104
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:94fc086f1cb5d90fd79d83048375736824c5d11a14a4cfe7a96b785301574ff4
3
  size 171641556
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2b2298a4d518354e12203d9c37823d7a652069287aa0e2102a95b05a590b5f3
3
  size 171641556