Hugging Face's logo

machinelearningzuu
/
queue_detection

Object Detection
Transformers
Safetensors
conditional_detr
Generated from Trainer
Inference Endpoints
Model card Files Community
machinelearningzuu commited on
Commit
155f78e
•
1 Parent(s): 4cfae30

End of training

Browse files
Files changed (2) hide show
  1. README.md +47 -19
  2. model.safetensors +1 -1
README.md CHANGED
@@ -15,23 +15,23 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [microsoft/conditional-detr-resnet-50](https://huggingface.co/microsoft/conditional-detr-resnet-50) on the None dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 1.4686
19
- - Map: 0.2556
20
- - Map 50: 0.4262
21
- - Map 75: 0.2687
22
  - Map Small: -1.0
23
- - Map Medium: 0.0006
24
- - Map Large: 0.2572
25
- - Mar 1: 0.2033
26
- - Mar 10: 0.561
27
- - Mar 100: 0.715
28
  - Mar Small: -1.0
29
- - Mar Medium: 0.0036
30
- - Mar Large: 0.7212
31
- - Map Cashier: 0.3957
32
- - Mar 100 Cashier: 0.812
33
- - Map Cx: 0.1154
34
- - Mar 100 Cx: 0.618
35
 
36
  ## Model description
37
 
@@ -56,20 +56,48 @@ The following hyperparameters were used during training:
56
  - seed: 42
57
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
58
  - lr_scheduler_type: cosine
59
- - num_epochs: 2
60
  - mixed_precision_training: Native AMP
61
 
62
  ### Training results
63
 
64
  | 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 Cashier | Mar 100 Cashier | Map Cx | Mar 100 Cx |
65
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:----------:|:---------:|:------:|:------:|:-------:|:---------:|:----------:|:---------:|:-----------:|:---------------:|:------:|:----------:|
66
- | No log | 1.0 | 218 | 1.6672 | 0.1422 | 0.2781 | 0.1441 | -1.0 | 0.0 | 0.1427 | 0.166 | 0.421 | 0.6116 | -1.0 | 0.0 | 0.6154 | 0.2345 | 0.7322 | 0.05 | 0.4911 |
67
- | No log | 2.0 | 436 | 1.4686 | 0.2556 | 0.4262 | 0.2687 | -1.0 | 0.0006 | 0.2572 | 0.2033 | 0.561 | 0.715 | -1.0 | 0.0036 | 0.7212 | 0.3957 | 0.812 | 0.1154 | 0.618 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
68
 
69
 
70
  ### Framework versions
71
 
72
  - Transformers 4.42.3
73
- - Pytorch 2.3.1+cpu
74
  - Datasets 2.20.0
75
  - Tokenizers 0.19.1
 
15
 
16
  This model is a fine-tuned version of [microsoft/conditional-detr-resnet-50](https://huggingface.co/microsoft/conditional-detr-resnet-50) on the None dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 1.4822
19
+ - Map: 0.2108
20
+ - Map 50: 0.3072
21
+ - Map 75: 0.2725
22
  - Map Small: -1.0
23
+ - Map Medium: -1.0
24
+ - Map Large: 0.2219
25
+ - Mar 1: 0.1833
26
+ - Mar 10: 0.4195
27
+ - Mar 100: 0.736
28
  - Mar Small: -1.0
29
+ - Mar Medium: -1.0
30
+ - Mar Large: 0.736
31
+ - Map Cashier: 0.0544
32
+ - Mar 100 Cashier: 0.8053
33
+ - Map Cx: 0.3672
34
+ - Mar 100 Cx: 0.6667
35
 
36
  ## Model description
37
 
 
56
  - seed: 42
57
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
58
  - lr_scheduler_type: cosine
59
+ - num_epochs: 30
60
  - mixed_precision_training: Native AMP
61
 
62
  ### Training results
63
 
64
  | 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 Cashier | Mar 100 Cashier | Map Cx | Mar 100 Cx |
65
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:----------:|:---------:|:------:|:------:|:-------:|:---------:|:----------:|:---------:|:-----------:|:---------------:|:------:|:----------:|
66
+ | No log | 1.0 | 5 | 38.9601 | 0.0007 | 0.0042 | 0.0 | -1.0 | -1.0 | 0.001 | 0.0 | 0.0186 | 0.053 | -1.0 | -1.0 | 0.053 | 0.0004 | 0.0526 | 0.001 | 0.0533 |
67
+ | No log | 2.0 | 10 | 20.3884 | 0.0053 | 0.0378 | 0.001 | -1.0 | -1.0 | 0.0119 | 0.0 | 0.0133 | 0.11 | -1.0 | -1.0 | 0.11 | 0.0 | 0.0 | 0.0106 | 0.22 |
68
+ | No log | 3.0 | 15 | 8.6229 | 0.0257 | 0.0582 | 0.0072 | -1.0 | -1.0 | 0.0573 | 0.0333 | 0.12 | 0.1737 | -1.0 | -1.0 | 0.1737 | 0.0002 | 0.0474 | 0.0512 | 0.3 |
69
+ | No log | 4.0 | 20 | 5.4986 | 0.0061 | 0.02 | 0.0018 | -1.0 | -1.0 | 0.0175 | 0.0 | 0.0833 | 0.1 | -1.0 | -1.0 | 0.1 | 0.0 | 0.0 | 0.0122 | 0.2 |
70
+ | No log | 5.0 | 25 | 3.2810 | 0.0152 | 0.0644 | 0.0 | -1.0 | -1.0 | 0.0182 | 0.0067 | 0.0933 | 0.1989 | -1.0 | -1.0 | 0.1989 | 0.004 | 0.1579 | 0.0264 | 0.24 |
71
+ | No log | 6.0 | 30 | 2.4527 | 0.0174 | 0.0707 | 0.0008 | -1.0 | -1.0 | 0.0201 | 0.0067 | 0.0667 | 0.3286 | -1.0 | -1.0 | 0.3286 | 0.0076 | 0.3105 | 0.0272 | 0.3467 |
72
+ | No log | 7.0 | 35 | 2.1059 | 0.0222 | 0.0715 | 0.0077 | -1.0 | 0.0 | 0.0244 | 0.0233 | 0.0993 | 0.4733 | -1.0 | 0.0 | 0.4857 | 0.017 | 0.6 | 0.0273 | 0.3467 |
73
+ | No log | 8.0 | 40 | 2.1013 | 0.035 | 0.117 | 0.0054 | -1.0 | -1.0 | 0.0354 | 0.0333 | 0.0467 | 0.3653 | -1.0 | -1.0 | 0.3653 | 0.0204 | 0.6105 | 0.0497 | 0.12 |
74
+ | No log | 9.0 | 45 | 1.9401 | 0.0412 | 0.1081 | 0.0058 | -1.0 | -1.0 | 0.042 | 0.0267 | 0.04 | 0.3639 | -1.0 | -1.0 | 0.3639 | 0.0184 | 0.6211 | 0.064 | 0.1067 |
75
+ | No log | 10.0 | 50 | 2.2227 | 0.0071 | 0.0254 | 0.0017 | -1.0 | -1.0 | 0.0075 | 0.0 | 0.0433 | 0.256 | -1.0 | -1.0 | 0.256 | 0.0044 | 0.3053 | 0.0098 | 0.2067 |
76
+ | No log | 11.0 | 55 | 1.9404 | 0.0126 | 0.033 | 0.01 | -1.0 | -1.0 | 0.0127 | 0.0 | 0.08 | 0.333 | -1.0 | -1.0 | 0.333 | 0.0082 | 0.4526 | 0.0169 | 0.2133 |
77
+ | No log | 12.0 | 60 | 1.8410 | 0.0186 | 0.0658 | 0.0078 | -1.0 | -1.0 | 0.0187 | 0.0 | 0.0733 | 0.3591 | -1.0 | -1.0 | 0.3591 | 0.0125 | 0.5316 | 0.0247 | 0.1867 |
78
+ | No log | 13.0 | 65 | 1.7476 | 0.0735 | 0.1829 | 0.0136 | -1.0 | -1.0 | 0.0736 | 0.0633 | 0.1133 | 0.4335 | -1.0 | -1.0 | 0.4335 | 0.0105 | 0.4737 | 0.1366 | 0.3933 |
79
+ | No log | 14.0 | 70 | 1.8939 | 0.0634 | 0.1492 | 0.0276 | -1.0 | -1.0 | 0.0638 | 0.05 | 0.2033 | 0.3719 | -1.0 | -1.0 | 0.3719 | 0.0032 | 0.2105 | 0.1236 | 0.5333 |
80
+ | No log | 15.0 | 75 | 1.7653 | 0.0535 | 0.1438 | 0.0387 | -1.0 | -1.0 | 0.0554 | 0.0964 | 0.1929 | 0.4917 | -1.0 | -1.0 | 0.4917 | 0.0103 | 0.4263 | 0.0968 | 0.5571 |
81
+ | No log | 16.0 | 80 | 1.6493 | 0.0988 | 0.2205 | 0.0787 | -1.0 | -1.0 | 0.1014 | 0.0733 | 0.2419 | 0.6505 | -1.0 | -1.0 | 0.6505 | 0.0225 | 0.7211 | 0.1751 | 0.58 |
82
+ | No log | 17.0 | 85 | 1.7624 | 0.1198 | 0.2464 | 0.0858 | 0.0 | -1.0 | 0.1289 | 0.1167 | 0.2333 | 0.6149 | 0.0 | -1.0 | 0.6352 | 0.0285 | 0.6632 | 0.211 | 0.5667 |
83
+ | No log | 18.0 | 90 | 1.8609 | 0.149 | 0.3181 | 0.0874 | -1.0 | -1.0 | 0.1517 | 0.13 | 0.2861 | 0.5789 | -1.0 | -1.0 | 0.5789 | 0.0403 | 0.6579 | 0.2576 | 0.5 |
84
+ | No log | 19.0 | 95 | 1.6860 | 0.1455 | 0.257 | 0.1026 | -1.0 | -1.0 | 0.1497 | 0.1393 | 0.2697 | 0.6496 | -1.0 | -1.0 | 0.6496 | 0.039 | 0.7421 | 0.2521 | 0.5571 |
85
+ | No log | 20.0 | 100 | 1.7541 | 0.1966 | 0.3274 | 0.2313 | -1.0 | -1.0 | 0.1994 | 0.17 | 0.317 | 0.6182 | -1.0 | -1.0 | 0.6182 | 0.0379 | 0.6632 | 0.3553 | 0.5733 |
86
+ | No log | 21.0 | 105 | 1.7043 | 0.1735 | 0.2572 | 0.1786 | -1.0 | 0.0 | 0.1913 | 0.1821 | 0.3295 | 0.6252 | -1.0 | 0.0 | 0.6472 | 0.0459 | 0.6789 | 0.3012 | 0.5714 |
87
+ | No log | 22.0 | 110 | 1.5489 | 0.1959 | 0.3051 | 0.264 | -1.0 | -1.0 | 0.2039 | 0.17 | 0.3568 | 0.6896 | -1.0 | -1.0 | 0.6896 | 0.046 | 0.7526 | 0.3458 | 0.6267 |
88
+ | No log | 23.0 | 115 | 1.6402 | 0.166 | 0.293 | 0.1791 | -1.0 | -1.0 | 0.1773 | 0.1432 | 0.2851 | 0.6496 | -1.0 | -1.0 | 0.6496 | 0.048 | 0.7526 | 0.284 | 0.5467 |
89
+ | No log | 24.0 | 120 | 1.5800 | 0.188 | 0.3099 | 0.1815 | -1.0 | -1.0 | 0.1965 | 0.1791 | 0.3337 | 0.693 | -1.0 | -1.0 | 0.693 | 0.0408 | 0.7526 | 0.3352 | 0.6333 |
90
+ | No log | 25.0 | 125 | 1.5566 | 0.1921 | 0.3038 | 0.2227 | -1.0 | -1.0 | 0.2037 | 0.17 | 0.4244 | 0.6928 | -1.0 | -1.0 | 0.6928 | 0.0642 | 0.7789 | 0.32 | 0.6067 |
91
+ | No log | 26.0 | 130 | 1.7227 | 0.2044 | 0.3337 | 0.2155 | -1.0 | -1.0 | 0.2145 | 0.1667 | 0.3319 | 0.5968 | -1.0 | -1.0 | 0.5968 | 0.0484 | 0.6737 | 0.3604 | 0.52 |
92
+ | No log | 27.0 | 135 | 1.5184 | 0.2095 | 0.3161 | 0.2389 | -1.0 | -1.0 | 0.2211 | 0.1877 | 0.3163 | 0.7026 | -1.0 | -1.0 | 0.7026 | 0.0603 | 0.8053 | 0.3586 | 0.6 |
93
+ | No log | 28.0 | 140 | 1.5156 | 0.2172 | 0.3273 | 0.2672 | -1.0 | -1.0 | 0.2286 | 0.2077 | 0.4402 | 0.7226 | -1.0 | -1.0 | 0.7226 | 0.0681 | 0.8053 | 0.3664 | 0.64 |
94
+ | No log | 29.0 | 145 | 1.6211 | 0.1652 | 0.2652 | 0.2007 | -1.0 | -1.0 | 0.1737 | 0.15 | 0.4081 | 0.677 | -1.0 | -1.0 | 0.677 | 0.0463 | 0.7474 | 0.2841 | 0.6067 |
95
+ | No log | 30.0 | 150 | 1.4822 | 0.2108 | 0.3072 | 0.2725 | -1.0 | -1.0 | 0.2219 | 0.1833 | 0.4195 | 0.736 | -1.0 | -1.0 | 0.736 | 0.0544 | 0.8053 | 0.3672 | 0.6667 |
96
 
97
 
98
  ### Framework versions
99
 
100
  - Transformers 4.42.3
101
+ - Pytorch 2.3.1+cu121
102
  - Datasets 2.20.0
103
  - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:87b97e4fb5d0f0501f5224be9f29f77a5fb391197b05506cee1fccf856f8f99b
3
  size 174076712
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c03221a8af7eac8b01b2f17f3be8f3b37468f54d37a0d8443e9476bfc99d789c
3
  size 174076712