Garon16 commited on
Commit
91fdc21
1 Parent(s): 79f8ad5

End of training

Browse files
README.md ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ language:
4
+ - ru
5
+ license: apache-2.0
6
+ base_model: PekingU/rtdetr_r50vd_coco_o365
7
+ tags:
8
+ - generated_from_trainer
9
+ model-index:
10
+ - name: RT-DETR Russian car plate detection with classification by type
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # RT-DETR Russian car plate detection with classification by type
18
+
19
+ This model is a fine-tuned version of [PekingU/rtdetr_r50vd_coco_o365](https://huggingface.co/PekingU/rtdetr_r50vd_coco_o365) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 4.1673
22
+ - Map: 0.8829
23
+ - Map 50: 0.9858
24
+ - Map 75: 0.9736
25
+ - Map Car-plates-and-these-types: -1.0
26
+ - Map Large: 0.9689
27
+ - Map Medium: 0.9125
28
+ - Map N P: 0.857
29
+ - Map P P: 0.9087
30
+ - Map Small: 0.696
31
+ - Mar 1: 0.8686
32
+ - Mar 10: 0.9299
33
+ - Mar 100: 0.9357
34
+ - Mar 100 Car-plates-and-these-types: -1.0
35
+ - Mar 100 N P: 0.9169
36
+ - Mar 100 P P: 0.9545
37
+ - Mar Large: 0.9844
38
+ - Mar Medium: 0.958
39
+ - Mar Small: 0.8354
40
+
41
+ ## Model description
42
+
43
+ More information needed
44
+
45
+ ## Intended uses & limitations
46
+
47
+ More information needed
48
+
49
+ ## Training and evaluation data
50
+
51
+ More information needed
52
+
53
+ ## Training procedure
54
+
55
+ ### Training hyperparameters
56
+
57
+ The following hyperparameters were used during training:
58
+ - learning_rate: 5e-05
59
+ - train_batch_size: 32
60
+ - eval_batch_size: 8
61
+ - seed: 42
62
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
63
+ - lr_scheduler_type: linear
64
+ - lr_scheduler_warmup_steps: 300
65
+ - num_epochs: 20
66
+
67
+ ### Training results
68
+
69
+ | Training Loss | Epoch | Step | Validation Loss | Map | Map 50 | Map 75 | Map Car-plates-and-these-types | Map Large | Map Medium | Map N P | Map P P | Map Small | Mar 1 | Mar 10 | Mar 100 | Mar 100 Car-plates-and-these-types | Mar 100 N P | Mar 100 P P | Mar Large | Mar Medium | Mar Small |
70
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:------------------------------:|:---------:|:----------:|:-------:|:-------:|:---------:|:------:|:------:|:-------:|:----------------------------------:|:-----------:|:-----------:|:---------:|:----------:|:---------:|
71
+ | No log | 1.0 | 109 | 64.6127 | 0.035 | 0.0558 | 0.0379 | -1.0 | 0.0039 | 0.0663 | 0.0191 | 0.0508 | 0.0071 | 0.1523 | 0.3009 | 0.3361 | -1.0 | 0.3179 | 0.3543 | 0.7625 | 0.3788 | 0.1157 |
72
+ | No log | 2.0 | 218 | 15.4008 | 0.8237 | 0.9418 | 0.9327 | -1.0 | 0.893 | 0.879 | 0.7945 | 0.8529 | 0.4319 | 0.8203 | 0.8924 | 0.9018 | -1.0 | 0.8766 | 0.9269 | 0.9656 | 0.9324 | 0.7653 |
73
+ | No log | 3.0 | 327 | 9.4050 | 0.8439 | 0.9566 | 0.9479 | -1.0 | 0.9439 | 0.8908 | 0.8158 | 0.872 | 0.5171 | 0.8416 | 0.908 | 0.9144 | -1.0 | 0.9002 | 0.9286 | 0.9781 | 0.9368 | 0.8051 |
74
+ | No log | 4.0 | 436 | 7.9164 | 0.8493 | 0.9665 | 0.9543 | -1.0 | 0.9567 | 0.8903 | 0.8338 | 0.8648 | 0.5581 | 0.8481 | 0.9159 | 0.9267 | -1.0 | 0.9173 | 0.936 | 0.975 | 0.949 | 0.8185 |
75
+ | 70.2867 | 5.0 | 545 | 6.8177 | 0.8525 | 0.9723 | 0.9602 | -1.0 | 0.9521 | 0.8918 | 0.8234 | 0.8816 | 0.6025 | 0.8438 | 0.9214 | 0.9279 | -1.0 | 0.9181 | 0.9378 | 0.975 | 0.9492 | 0.8211 |
76
+ | 70.2867 | 6.0 | 654 | 6.0182 | 0.854 | 0.9744 | 0.9619 | -1.0 | 0.9574 | 0.8912 | 0.8251 | 0.8829 | 0.6123 | 0.8438 | 0.9176 | 0.927 | -1.0 | 0.9137 | 0.9403 | 0.9781 | 0.9503 | 0.8163 |
77
+ | 70.2867 | 7.0 | 763 | 5.4024 | 0.8731 | 0.9772 | 0.9667 | -1.0 | 0.9635 | 0.9113 | 0.8462 | 0.9001 | 0.6376 | 0.8608 | 0.9275 | 0.9336 | -1.0 | 0.9202 | 0.9471 | 0.9781 | 0.956 | 0.8266 |
78
+ | 70.2867 | 8.0 | 872 | 5.2224 | 0.8726 | 0.9809 | 0.9767 | -1.0 | 0.9582 | 0.9069 | 0.8487 | 0.8966 | 0.6472 | 0.8625 | 0.9265 | 0.9301 | -1.0 | 0.9137 | 0.9464 | 0.9875 | 0.9528 | 0.8232 |
79
+ | 70.2867 | 9.0 | 981 | 4.7844 | 0.8679 | 0.9821 | 0.9687 | -1.0 | 0.9574 | 0.9023 | 0.8451 | 0.8907 | 0.6382 | 0.8606 | 0.9213 | 0.9283 | -1.0 | 0.9119 | 0.9448 | 0.9844 | 0.952 | 0.8165 |
80
+ | 4.2466 | 10.0 | 1090 | 5.1437 | 0.8729 | 0.9816 | 0.9762 | -1.0 | 0.9577 | 0.9028 | 0.8448 | 0.901 | 0.6686 | 0.8605 | 0.9296 | 0.9359 | -1.0 | 0.9203 | 0.9514 | 0.9781 | 0.9567 | 0.8413 |
81
+ | 4.2466 | 11.0 | 1199 | 4.5169 | 0.8858 | 0.9828 | 0.9768 | -1.0 | 0.9707 | 0.9162 | 0.8628 | 0.9087 | 0.6734 | 0.8695 | 0.9264 | 0.931 | -1.0 | 0.9121 | 0.95 | 0.9781 | 0.9538 | 0.823 |
82
+ | 4.2466 | 12.0 | 1308 | 4.5858 | 0.8813 | 0.9865 | 0.9744 | -1.0 | 0.9623 | 0.9126 | 0.8585 | 0.9041 | 0.6815 | 0.8671 | 0.9308 | 0.9355 | -1.0 | 0.9185 | 0.9526 | 0.9812 | 0.9583 | 0.8308 |
83
+ | 4.2466 | 13.0 | 1417 | 4.5345 | 0.8778 | 0.9843 | 0.9726 | -1.0 | 0.957 | 0.9101 | 0.8526 | 0.903 | 0.6754 | 0.8628 | 0.9281 | 0.9335 | -1.0 | 0.9158 | 0.9512 | 0.9812 | 0.9557 | 0.8314 |
84
+ | 3.589 | 14.0 | 1526 | 4.3003 | 0.8885 | 0.9857 | 0.9759 | -1.0 | 0.9656 | 0.9189 | 0.8642 | 0.9128 | 0.6957 | 0.8724 | 0.9334 | 0.9375 | -1.0 | 0.9194 | 0.9555 | 0.9875 | 0.959 | 0.8375 |
85
+ | 3.589 | 15.0 | 1635 | 4.3999 | 0.8819 | 0.986 | 0.9741 | -1.0 | 0.9606 | 0.9118 | 0.8575 | 0.9064 | 0.6892 | 0.8659 | 0.9283 | 0.9336 | -1.0 | 0.9137 | 0.9534 | 0.9844 | 0.9566 | 0.8245 |
86
+ | 3.589 | 16.0 | 1744 | 4.2719 | 0.8796 | 0.986 | 0.9726 | -1.0 | 0.9661 | 0.9093 | 0.8543 | 0.905 | 0.6914 | 0.8649 | 0.927 | 0.9313 | -1.0 | 0.9121 | 0.9505 | 0.9875 | 0.9543 | 0.8266 |
87
+ | 3.589 | 17.0 | 1853 | 4.2497 | 0.8838 | 0.9845 | 0.9733 | -1.0 | 0.9656 | 0.9141 | 0.8599 | 0.9077 | 0.6997 | 0.8678 | 0.9295 | 0.9352 | -1.0 | 0.9141 | 0.9562 | 0.9812 | 0.958 | 0.832 |
88
+ | 3.589 | 18.0 | 1962 | 4.2807 | 0.8829 | 0.9855 | 0.9754 | -1.0 | 0.9673 | 0.9121 | 0.8558 | 0.9099 | 0.6964 | 0.8683 | 0.9286 | 0.9337 | -1.0 | 0.9126 | 0.9548 | 0.9844 | 0.9555 | 0.8357 |
89
+ | 3.2442 | 19.0 | 2071 | 4.1978 | 0.8835 | 0.9861 | 0.9748 | -1.0 | 0.9675 | 0.9121 | 0.8559 | 0.911 | 0.6932 | 0.8691 | 0.9272 | 0.9336 | -1.0 | 0.9134 | 0.9538 | 0.9844 | 0.9557 | 0.8337 |
90
+ | 3.2442 | 20.0 | 2180 | 4.1673 | 0.8829 | 0.9858 | 0.9736 | -1.0 | 0.9689 | 0.9125 | 0.857 | 0.9087 | 0.696 | 0.8686 | 0.9299 | 0.9357 | -1.0 | 0.9169 | 0.9545 | 0.9844 | 0.958 | 0.8354 |
91
+
92
+
93
+ ### Framework versions
94
+
95
+ - Transformers 4.46.0.dev0
96
+ - Pytorch 2.5.0+cu124
97
+ - Tokenizers 0.20.1
config.json ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "PekingU/rtdetr_r50vd_coco_o365",
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "silu",
5
+ "anchor_image_size": null,
6
+ "architectures": [
7
+ "RTDetrForObjectDetection"
8
+ ],
9
+ "attention_dropout": 0.0,
10
+ "auxiliary_loss": true,
11
+ "backbone": null,
12
+ "backbone_config": {
13
+ "model_type": "rt_detr_resnet",
14
+ "out_features": [
15
+ "stage2",
16
+ "stage3",
17
+ "stage4"
18
+ ],
19
+ "out_indices": [
20
+ 2,
21
+ 3,
22
+ 4
23
+ ]
24
+ },
25
+ "backbone_kwargs": null,
26
+ "batch_norm_eps": 1e-05,
27
+ "box_noise_scale": 1.0,
28
+ "d_model": 256,
29
+ "decoder_activation_function": "relu",
30
+ "decoder_attention_heads": 8,
31
+ "decoder_ffn_dim": 1024,
32
+ "decoder_in_channels": [
33
+ 256,
34
+ 256,
35
+ 256
36
+ ],
37
+ "decoder_layers": 6,
38
+ "decoder_n_points": 4,
39
+ "disable_custom_kernels": true,
40
+ "dropout": 0.0,
41
+ "encode_proj_layers": [
42
+ 2
43
+ ],
44
+ "encoder_activation_function": "gelu",
45
+ "encoder_attention_heads": 8,
46
+ "encoder_ffn_dim": 1024,
47
+ "encoder_hidden_dim": 256,
48
+ "encoder_in_channels": [
49
+ 512,
50
+ 1024,
51
+ 2048
52
+ ],
53
+ "encoder_layers": 1,
54
+ "eos_coefficient": 0.0001,
55
+ "eval_size": null,
56
+ "feat_strides": [
57
+ 8,
58
+ 16,
59
+ 32
60
+ ],
61
+ "focal_loss_alpha": 0.75,
62
+ "focal_loss_gamma": 2.0,
63
+ "freeze_backbone_batch_norms": true,
64
+ "hidden_expansion": 1.0,
65
+ "id2label": {
66
+ "0": "car-plates-and-these-types",
67
+ "1": "n_p",
68
+ "2": "p_p"
69
+ },
70
+ "initializer_bias_prior_prob": null,
71
+ "initializer_range": 0.01,
72
+ "is_encoder_decoder": true,
73
+ "label2id": {
74
+ "car-plates-and-these-types": 0,
75
+ "n_p": 1,
76
+ "p_p": 2
77
+ },
78
+ "label_noise_ratio": 0.5,
79
+ "layer_norm_eps": 1e-05,
80
+ "learn_initial_query": false,
81
+ "matcher_alpha": 0.25,
82
+ "matcher_bbox_cost": 5.0,
83
+ "matcher_class_cost": 2.0,
84
+ "matcher_gamma": 2.0,
85
+ "matcher_giou_cost": 2.0,
86
+ "model_type": "rt_detr",
87
+ "normalize_before": false,
88
+ "num_denoising": 100,
89
+ "num_feature_levels": 3,
90
+ "num_queries": 300,
91
+ "positional_encoding_temperature": 10000,
92
+ "torch_dtype": "float32",
93
+ "transformers_version": "4.46.0.dev0",
94
+ "use_focal_loss": true,
95
+ "use_pretrained_backbone": false,
96
+ "use_timm_backbone": false,
97
+ "weight_loss_bbox": 5.0,
98
+ "weight_loss_giou": 2.0,
99
+ "weight_loss_vfl": 1.0,
100
+ "with_box_refine": true
101
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b249dcc7b4d41237731f8310c7e4918861bde396f11fe85a9b62b7e13ad2aae
3
+ size 171542900
preprocessor_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_annotations": true,
3
+ "do_normalize": false,
4
+ "do_pad": false,
5
+ "do_rescale": true,
6
+ "do_resize": true,
7
+ "format": "coco_detection",
8
+ "image_mean": [
9
+ 0.485,
10
+ 0.456,
11
+ 0.406
12
+ ],
13
+ "image_processor_type": "RTDetrImageProcessor",
14
+ "image_std": [
15
+ 0.229,
16
+ 0.224,
17
+ 0.225
18
+ ],
19
+ "pad_size": null,
20
+ "resample": 2,
21
+ "rescale_factor": 0.00392156862745098,
22
+ "size": {
23
+ "height": 640,
24
+ "width": 640
25
+ }
26
+ }
runs/Oct23_22-50-00_rentserver/events.out.tfevents.1729713001.rentserver.3359.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8fcc9dd86869dee233ca9a8e9255d46b2e2460be3fb8c4d2df9bbe54beffa13d
3
+ size 6645
runs/Oct23_22-51-12_rentserver/events.out.tfevents.1729713072.rentserver.3748.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:751904cbadf23197ebd00658cb7a13cd152d6a815d58271047d3b27614a4323c
3
+ size 32860
runs/Oct23_23-42-32_rentserver/events.out.tfevents.1729716152.rentserver.12988.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:319668d54dfbf14c919b8a70735db434e01fde87e13c0bef2fde2e136f263601
3
+ size 6718
runs/Oct23_23-43-42_rentserver/events.out.tfevents.1729716222.rentserver.13386.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32965d65aa437d6df96482b45aeb6629d546f231990be11a91a334ce577f08da
3
+ size 6702
runs/Oct23_23-45-17_rentserver/events.out.tfevents.1729716318.rentserver.13850.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ce6a924a7e701f62f6b91be0bcb66ab045383e569901a4f8fb1ce6069151ce7
3
+ size 6999
runs/Oct23_23-46-30_rentserver/events.out.tfevents.1729716391.rentserver.14116.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:603abf58ecc577a3e84905bcb9805ad776587250d217e04dce40240416d56b9d
3
+ size 6999
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ccc55adefb756fb44fa9f784744962d544b271272ad56a7af25c60c05d65a80
3
+ size 5240