Hugging Face's logo

qubvel-hf
/
debug_no_pad

Object Detection
Transformers
Safetensors
rt_detr
Generated from Trainer
Inference Endpoints
Model card Files Community
qubvel-hf HF staff commited on
Commit
c13cf64
•
1 Parent(s): 7ef2cb5

End of training

Browse files
Files changed (5) hide show
  1. README.md +108 -0
  2. config.json +101 -0
  3. model.safetensors +3 -0
  4. preprocessor_config.json +45 -0
  5. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: sbchoi/rtdetr_r50vd_coco_o365
3
+ tags:
4
+ - generated_from_trainer
5
+ model-index:
6
+ - name: debug_no_pad
7
+ results: []
8
+ ---
9
+
10
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
11
+ should probably proofread and complete it, then remove this comment. -->
12
+
13
+ # debug_no_pad
14
+
15
+ This model is a fine-tuned version of [sbchoi/rtdetr_r50vd_coco_o365](https://huggingface.co/sbchoi/rtdetr_r50vd_coco_o365) on an unknown dataset.
16
+ It achieves the following results on the evaluation set:
17
+ - Loss: 13.1506
18
+ - Map: 0.2026
19
+ - Map 50: 0.4492
20
+ - Map 75: 0.1638
21
+ - Map Small: 0.0726
22
+ - Map Medium: 0.1655
23
+ - Map Large: 0.2732
24
+ - Mar 1: 0.2221
25
+ - Mar 10: 0.3782
26
+ - Mar 100: 0.4312
27
+ - Mar Small: 0.1892
28
+ - Mar Medium: 0.3446
29
+ - Mar Large: 0.5587
30
+ - Map Coverall: 0.2662
31
+ - Mar 100 Coverall: 0.5721
32
+ - Map Face Shield: 0.1827
33
+ - Mar 100 Face Shield: 0.519
34
+ - Map Gloves: 0.2317
35
+ - Mar 100 Gloves: 0.3991
36
+ - Map Goggles: 0.1477
37
+ - Mar 100 Goggles: 0.2938
38
+ - Map Mask: 0.1846
39
+ - Mar 100 Mask: 0.372
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: 0.0001
59
+ - train_batch_size: 8
60
+ - eval_batch_size: 8
61
+ - seed: 42
62
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
63
+ - lr_scheduler_type: linear
64
+ - lr_scheduler_warmup_steps: 200
65
+ - num_epochs: 30
66
+
67
+ ### Training results
68
+
69
+ | 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 Coverall | Mar 100 Coverall | Map Face Shield | Mar 100 Face Shield | Map Gloves | Mar 100 Gloves | Map Goggles | Mar 100 Goggles | Map Mask | Mar 100 Mask |
70
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:----------:|:---------:|:------:|:------:|:-------:|:---------:|:----------:|:---------:|:------------:|:----------------:|:---------------:|:-------------------:|:----------:|:--------------:|:-----------:|:---------------:|:--------:|:------------:|
71
+ | No log | 1.0 | 107 | 101.3493 | 0.0 | 0.0002 | 0.0 | 0.0 | 0.0 | 0.0004 | 0.0026 | 0.0098 | 0.0201 | 0.0144 | 0.0106 | 0.0257 | 0.0002 | 0.0387 | 0.0 | 0.0139 | 0.0 | 0.0018 | 0.0 | 0.0462 | 0.0 | 0.0 |
72
+ | No log | 2.0 | 214 | 29.9261 | 0.0003 | 0.0014 | 0.0 | 0.0 | 0.0 | 0.0003 | 0.0019 | 0.0114 | 0.0182 | 0.0 | 0.0133 | 0.0184 | 0.0013 | 0.091 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 |
73
+ | No log | 3.0 | 321 | 21.9254 | 0.0031 | 0.0095 | 0.0014 | 0.0023 | 0.008 | 0.003 | 0.0233 | 0.0687 | 0.1387 | 0.0058 | 0.0927 | 0.1278 | 0.0119 | 0.3468 | 0.0007 | 0.1494 | 0.0009 | 0.0281 | 0.0 | 0.0308 | 0.0018 | 0.1382 |
74
+ | No log | 4.0 | 428 | 17.4624 | 0.0317 | 0.0912 | 0.0132 | 0.0206 | 0.0307 | 0.0303 | 0.0537 | 0.1415 | 0.2005 | 0.0667 | 0.1778 | 0.2273 | 0.0349 | 0.3063 | 0.0077 | 0.2443 | 0.0105 | 0.1442 | 0.0012 | 0.0754 | 0.1044 | 0.2324 |
75
+ | 97.4578 | 5.0 | 535 | 16.2654 | 0.0654 | 0.1715 | 0.0386 | 0.031 | 0.0647 | 0.0918 | 0.1116 | 0.2112 | 0.2733 | 0.0983 | 0.2249 | 0.358 | 0.0664 | 0.445 | 0.0501 | 0.3405 | 0.0282 | 0.1335 | 0.031 | 0.1692 | 0.151 | 0.2782 |
76
+ | 97.4578 | 6.0 | 642 | 15.4886 | 0.0997 | 0.2444 | 0.0711 | 0.0568 | 0.0907 | 0.1367 | 0.1635 | 0.2746 | 0.3398 | 0.1459 | 0.2877 | 0.4549 | 0.114 | 0.4734 | 0.0637 | 0.3797 | 0.0467 | 0.2299 | 0.0762 | 0.2662 | 0.1982 | 0.3498 |
77
+ | 97.4578 | 7.0 | 749 | 15.3647 | 0.1205 | 0.3157 | 0.0659 | 0.0501 | 0.1134 | 0.1725 | 0.1602 | 0.2869 | 0.3296 | 0.1181 | 0.2841 | 0.4334 | 0.1443 | 0.3901 | 0.0689 | 0.4139 | 0.1082 | 0.2714 | 0.0928 | 0.2523 | 0.1882 | 0.3204 |
78
+ | 97.4578 | 8.0 | 856 | 15.2431 | 0.1016 | 0.2715 | 0.0558 | 0.0301 | 0.0905 | 0.146 | 0.1569 | 0.2829 | 0.344 | 0.129 | 0.2983 | 0.4391 | 0.1136 | 0.4009 | 0.0707 | 0.438 | 0.0902 | 0.2915 | 0.0755 | 0.2923 | 0.1579 | 0.2973 |
79
+ | 97.4578 | 9.0 | 963 | 14.7396 | 0.1497 | 0.3629 | 0.1013 | 0.0641 | 0.1617 | 0.1877 | 0.1691 | 0.3051 | 0.3433 | 0.2003 | 0.3204 | 0.4348 | 0.1987 | 0.4293 | 0.1105 | 0.3861 | 0.1467 | 0.3379 | 0.1034 | 0.2462 | 0.1894 | 0.3169 |
80
+ | 27.2963 | 10.0 | 1070 | 14.2373 | 0.1549 | 0.3715 | 0.1124 | 0.0798 | 0.163 | 0.1902 | 0.192 | 0.3368 | 0.4073 | 0.2642 | 0.3594 | 0.5035 | 0.1891 | 0.4977 | 0.1105 | 0.4646 | 0.1507 | 0.3763 | 0.1091 | 0.3108 | 0.2154 | 0.3871 |
81
+ | 27.2963 | 11.0 | 1177 | 14.4769 | 0.1413 | 0.3176 | 0.1085 | 0.0477 | 0.1304 | 0.2084 | 0.1892 | 0.3246 | 0.3797 | 0.195 | 0.3323 | 0.4877 | 0.1858 | 0.4775 | 0.1162 | 0.4532 | 0.1387 | 0.3683 | 0.1182 | 0.2569 | 0.1477 | 0.3427 |
82
+ | 27.2963 | 12.0 | 1284 | 13.9935 | 0.1922 | 0.4287 | 0.146 | 0.0793 | 0.1931 | 0.2346 | 0.2099 | 0.3566 | 0.4036 | 0.2114 | 0.3472 | 0.5202 | 0.2718 | 0.5117 | 0.1478 | 0.4392 | 0.1856 | 0.3893 | 0.1322 | 0.3 | 0.2236 | 0.3778 |
83
+ | 27.2963 | 13.0 | 1391 | 13.7867 | 0.1745 | 0.4012 | 0.1326 | 0.0796 | 0.1727 | 0.2301 | 0.1965 | 0.3503 | 0.4054 | 0.2274 | 0.352 | 0.5241 | 0.2083 | 0.5086 | 0.1234 | 0.4633 | 0.2004 | 0.3924 | 0.1076 | 0.2754 | 0.2327 | 0.3871 |
84
+ | 27.2963 | 14.0 | 1498 | 13.7474 | 0.18 | 0.3989 | 0.1445 | 0.0579 | 0.1639 | 0.241 | 0.1995 | 0.3449 | 0.3898 | 0.2128 | 0.3156 | 0.5022 | 0.1854 | 0.4761 | 0.1402 | 0.4038 | 0.2065 | 0.4094 | 0.1273 | 0.2692 | 0.2409 | 0.3902 |
85
+ | 24.8642 | 15.0 | 1605 | 13.4012 | 0.1958 | 0.4352 | 0.157 | 0.0757 | 0.1788 | 0.2823 | 0.2073 | 0.3522 | 0.4008 | 0.212 | 0.3359 | 0.5287 | 0.198 | 0.5099 | 0.1506 | 0.4101 | 0.2134 | 0.3933 | 0.1558 | 0.2831 | 0.2611 | 0.4076 |
86
+ | 24.8642 | 16.0 | 1712 | 13.3569 | 0.19 | 0.4224 | 0.1454 | 0.1542 | 0.2044 | 0.2407 | 0.2128 | 0.3568 | 0.3989 | 0.2681 | 0.3554 | 0.5058 | 0.1902 | 0.4977 | 0.1678 | 0.3911 | 0.225 | 0.3978 | 0.1325 | 0.2923 | 0.2347 | 0.4156 |
87
+ | 24.8642 | 17.0 | 1819 | 13.4929 | 0.1809 | 0.3983 | 0.1353 | 0.0679 | 0.1746 | 0.267 | 0.2142 | 0.3496 | 0.3898 | 0.2683 | 0.3211 | 0.5047 | 0.2047 | 0.5185 | 0.174 | 0.4329 | 0.1679 | 0.321 | 0.1303 | 0.2769 | 0.2276 | 0.3996 |
88
+ | 24.8642 | 18.0 | 1926 | 13.4921 | 0.1789 | 0.3853 | 0.1477 | 0.057 | 0.1655 | 0.2511 | 0.2202 | 0.3641 | 0.4147 | 0.2614 | 0.349 | 0.5348 | 0.2266 | 0.55 | 0.1629 | 0.4759 | 0.2087 | 0.3857 | 0.138 | 0.2908 | 0.1584 | 0.3711 |
89
+ | 23.551 | 19.0 | 2033 | 13.3617 | 0.1824 | 0.3978 | 0.1471 | 0.0609 | 0.1695 | 0.2607 | 0.2263 | 0.388 | 0.4319 | 0.2304 | 0.3643 | 0.555 | 0.2282 | 0.5536 | 0.1568 | 0.4886 | 0.2076 | 0.4094 | 0.1415 | 0.3046 | 0.1776 | 0.4031 |
90
+ | 23.551 | 20.0 | 2140 | 13.3499 | 0.1834 | 0.4074 | 0.1496 | 0.0734 | 0.156 | 0.2478 | 0.2276 | 0.3903 | 0.4373 | 0.2858 | 0.3555 | 0.5568 | 0.2326 | 0.5392 | 0.1721 | 0.4987 | 0.2169 | 0.4237 | 0.1402 | 0.3508 | 0.1553 | 0.3742 |
91
+ | 23.551 | 21.0 | 2247 | 13.4009 | 0.1858 | 0.3991 | 0.1394 | 0.0578 | 0.1748 | 0.2553 | 0.2214 | 0.375 | 0.4254 | 0.1951 | 0.3547 | 0.5506 | 0.2349 | 0.5473 | 0.1842 | 0.4861 | 0.2227 | 0.4223 | 0.1443 | 0.3185 | 0.1428 | 0.3529 |
92
+ | 23.551 | 22.0 | 2354 | 13.4129 | 0.1824 | 0.3995 | 0.1459 | 0.0715 | 0.1394 | 0.2581 | 0.2234 | 0.3735 | 0.4277 | 0.2243 | 0.348 | 0.5472 | 0.2425 | 0.5491 | 0.1465 | 0.4949 | 0.2072 | 0.417 | 0.133 | 0.3015 | 0.1826 | 0.376 |
93
+ | 23.551 | 23.0 | 2461 | 13.4100 | 0.1902 | 0.4141 | 0.1602 | 0.0641 | 0.171 | 0.2609 | 0.2162 | 0.3732 | 0.4327 | 0.2437 | 0.368 | 0.5519 | 0.2405 | 0.5644 | 0.1554 | 0.5139 | 0.2313 | 0.4071 | 0.1506 | 0.3185 | 0.1733 | 0.3596 |
94
+ | 23.8193 | 24.0 | 2568 | 13.3091 | 0.1857 | 0.4151 | 0.1452 | 0.0708 | 0.1669 | 0.2486 | 0.214 | 0.3653 | 0.4232 | 0.2201 | 0.358 | 0.5354 | 0.2348 | 0.5676 | 0.145 | 0.4899 | 0.2294 | 0.4004 | 0.1476 | 0.2938 | 0.1717 | 0.3644 |
95
+ | 23.8193 | 25.0 | 2675 | 13.2781 | 0.2006 | 0.4435 | 0.1611 | 0.065 | 0.1657 | 0.2624 | 0.215 | 0.3741 | 0.4223 | 0.1951 | 0.351 | 0.5465 | 0.2609 | 0.5595 | 0.1831 | 0.5025 | 0.2365 | 0.3982 | 0.1364 | 0.2877 | 0.1863 | 0.3636 |
96
+ | 23.8193 | 26.0 | 2782 | 13.2183 | 0.1951 | 0.4333 | 0.1577 | 0.063 | 0.1709 | 0.2501 | 0.22 | 0.3751 | 0.4286 | 0.1712 | 0.3619 | 0.5411 | 0.2431 | 0.5734 | 0.163 | 0.4772 | 0.2165 | 0.4013 | 0.1581 | 0.3215 | 0.1946 | 0.3693 |
97
+ | 23.8193 | 27.0 | 2889 | 13.2704 | 0.2009 | 0.4453 | 0.1559 | 0.0626 | 0.1881 | 0.2605 | 0.2209 | 0.3766 | 0.4293 | 0.1594 | 0.371 | 0.5524 | 0.2635 | 0.5721 | 0.1797 | 0.5127 | 0.2159 | 0.4031 | 0.159 | 0.2938 | 0.1864 | 0.3649 |
98
+ | 23.8193 | 28.0 | 2996 | 13.1710 | 0.2062 | 0.4625 | 0.1583 | 0.0722 | 0.1803 | 0.2726 | 0.2227 | 0.3843 | 0.433 | 0.2041 | 0.3729 | 0.5589 | 0.2826 | 0.5838 | 0.1861 | 0.5101 | 0.2225 | 0.3973 | 0.1413 | 0.3015 | 0.1986 | 0.372 |
99
+ | 23.7672 | 29.0 | 3103 | 13.1248 | 0.2038 | 0.4521 | 0.1679 | 0.0703 | 0.1655 | 0.2771 | 0.2255 | 0.3828 | 0.4367 | 0.1904 | 0.3678 | 0.5661 | 0.2629 | 0.5757 | 0.1836 | 0.5241 | 0.2332 | 0.4027 | 0.1466 | 0.3031 | 0.1928 | 0.3778 |
100
+ | 23.7672 | 30.0 | 3210 | 13.1506 | 0.2026 | 0.4492 | 0.1638 | 0.0726 | 0.1655 | 0.2732 | 0.2221 | 0.3782 | 0.4312 | 0.1892 | 0.3446 | 0.5587 | 0.2662 | 0.5721 | 0.1827 | 0.519 | 0.2317 | 0.3991 | 0.1477 | 0.2938 | 0.1846 | 0.372 |
101
+
102
+
103
+ ### Framework versions
104
+
105
+ - Transformers 4.42.0.dev0
106
+ - Pytorch 2.1.0+cu118
107
+ - Datasets 2.19.1
108
+ - Tokenizers 0.19.1
config.json ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "sbchoi/rtdetr_r50vd_coco_o365",
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "silu",
5
+ "anchor_image_size": [
6
+ 640,
7
+ 640
8
+ ],
9
+ "architectures": [
10
+ "RTDetrForObjectDetection"
11
+ ],
12
+ "attention_dropout": 0.0,
13
+ "auxiliary_loss": true,
14
+ "backbone": "resnet50d",
15
+ "backbone_config": null,
16
+ "backbone_kwargs": {
17
+ "features_only": true,
18
+ "out_indices": [
19
+ 2,
20
+ 3,
21
+ 4
22
+ ]
23
+ },
24
+ "batch_norm_eps": 1e-05,
25
+ "box_noise_scale": 1.0,
26
+ "d_model": 256,
27
+ "decoder_activation_function": "relu",
28
+ "decoder_attention_heads": 8,
29
+ "decoder_ffn_dim": 1024,
30
+ "decoder_in_channels": [
31
+ 256,
32
+ 256,
33
+ 256
34
+ ],
35
+ "decoder_layers": 6,
36
+ "decoder_n_points": 4,
37
+ "disable_custom_kernels": true,
38
+ "dropout": 0.0,
39
+ "encode_proj_layers": [
40
+ 2
41
+ ],
42
+ "encoder_activation_function": "gelu",
43
+ "encoder_attention_heads": 8,
44
+ "encoder_ffn_dim": 1024,
45
+ "encoder_hidden_dim": 256,
46
+ "encoder_in_channels": [
47
+ 512,
48
+ 1024,
49
+ 2048
50
+ ],
51
+ "encoder_layers": 1,
52
+ "eos_coefficient": 0.0001,
53
+ "eval_size": null,
54
+ "feat_strides": [
55
+ 8,
56
+ 16,
57
+ 32
58
+ ],
59
+ "focal_loss_alpha": 0.75,
60
+ "focal_loss_gamma": 2.0,
61
+ "hidden_expansion": 1.0,
62
+ "id2label": {
63
+ "0": "Coverall",
64
+ "1": "Face_Shield",
65
+ "2": "Gloves",
66
+ "3": "Goggles",
67
+ "4": "Mask"
68
+ },
69
+ "initializer_range": 0.01,
70
+ "is_encoder_decoder": true,
71
+ "label2id": {
72
+ "Coverall": 0,
73
+ "Face_Shield": 1,
74
+ "Gloves": 2,
75
+ "Goggles": 3,
76
+ "Mask": 4
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.42.0.dev0",
94
+ "use_focal_loss": true,
95
+ "use_pretrained_backbone": true,
96
+ "use_timm_backbone": true,
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:47b6117c5e3ca2f5b6a449bd947240ec2439649a0e918c31c7d0947baac16fa1
3
+ size 171552884
preprocessor_config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_valid_processor_keys": [
3
+ "images",
4
+ "annotations",
5
+ "return_segmentation_masks",
6
+ "masks_path",
7
+ "do_resize",
8
+ "size",
9
+ "resample",
10
+ "do_rescale",
11
+ "rescale_factor",
12
+ "do_normalize",
13
+ "do_convert_annotations",
14
+ "image_mean",
15
+ "image_std",
16
+ "do_pad",
17
+ "format",
18
+ "return_tensors",
19
+ "data_format",
20
+ "input_data_format"
21
+ ],
22
+ "do_convert_annotations": true,
23
+ "do_normalize": false,
24
+ "do_pad": false,
25
+ "do_rescale": true,
26
+ "do_resize": true,
27
+ "format": "coco_detection",
28
+ "image_mean": [
29
+ 0.485,
30
+ 0.456,
31
+ 0.406
32
+ ],
33
+ "image_processor_type": "RTDetrImageProcessor",
34
+ "image_std": [
35
+ 0.229,
36
+ 0.224,
37
+ 0.225
38
+ ],
39
+ "resample": 2,
40
+ "rescale_factor": 0.00392156862745098,
41
+ "size": {
42
+ "height": 640,
43
+ "width": 640
44
+ }
45
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d415ca6d7250ef6ffcfe0059c6288b51fc97a1b2db8b847c1d308c998be97ea9
3
+ size 5112