End of training
Browse files- README.md +97 -0
- config.json +101 -0
- model.safetensors +3 -0
- preprocessor_config.json +26 -0
- runs/Oct23_22-50-00_rentserver/events.out.tfevents.1729713001.rentserver.3359.0 +3 -0
- runs/Oct23_22-51-12_rentserver/events.out.tfevents.1729713072.rentserver.3748.0 +3 -0
- runs/Oct23_23-42-32_rentserver/events.out.tfevents.1729716152.rentserver.12988.0 +3 -0
- runs/Oct23_23-43-42_rentserver/events.out.tfevents.1729716222.rentserver.13386.0 +3 -0
- runs/Oct23_23-45-17_rentserver/events.out.tfevents.1729716318.rentserver.13850.0 +3 -0
- runs/Oct23_23-46-30_rentserver/events.out.tfevents.1729716391.rentserver.14116.0 +3 -0
- training_args.bin +3 -0
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
|