End of training
Browse files- README.md +108 -0
- model.safetensors +1 -1
README.md
ADDED
|
@@ -0,0 +1,108 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
base_model: microsoft/conditional-detr-resnet-50
|
| 4 |
+
tags:
|
| 5 |
+
- generated_from_trainer
|
| 6 |
+
model-index:
|
| 7 |
+
- name: detr_finetuned_cppe5
|
| 8 |
+
results: []
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
| 12 |
+
should probably proofread and complete it, then remove this comment. -->
|
| 13 |
+
|
| 14 |
+
# detr_finetuned_cppe5
|
| 15 |
+
|
| 16 |
+
This model is a fine-tuned version of [microsoft/conditional-detr-resnet-50](https://huggingface.co/microsoft/conditional-detr-resnet-50) on an unknown dataset.
|
| 17 |
+
It achieves the following results on the evaluation set:
|
| 18 |
+
- Loss: 1.3461
|
| 19 |
+
- Map: 0.2811
|
| 20 |
+
- Map 50: 0.561
|
| 21 |
+
- Map 75: 0.239
|
| 22 |
+
- Map Small: 0.0945
|
| 23 |
+
- Map Medium: 0.2317
|
| 24 |
+
- Map Large: 0.419
|
| 25 |
+
- Mar 1: 0.2736
|
| 26 |
+
- Mar 10: 0.4208
|
| 27 |
+
- Mar 100: 0.4388
|
| 28 |
+
- Mar Small: 0.2191
|
| 29 |
+
- Mar Medium: 0.3871
|
| 30 |
+
- Mar Large: 0.598
|
| 31 |
+
- Map Coverall: 0.5394
|
| 32 |
+
- Mar 100 Coverall: 0.6554
|
| 33 |
+
- Map Face Shield: 0.2284
|
| 34 |
+
- Mar 100 Face Shield: 0.4405
|
| 35 |
+
- Map Gloves: 0.1791
|
| 36 |
+
- Mar 100 Gloves: 0.3598
|
| 37 |
+
- Map Goggles: 0.1786
|
| 38 |
+
- Mar 100 Goggles: 0.3354
|
| 39 |
+
- Map Mask: 0.2802
|
| 40 |
+
- Mar 100 Mask: 0.4027
|
| 41 |
+
|
| 42 |
+
## Model description
|
| 43 |
+
|
| 44 |
+
More information needed
|
| 45 |
+
|
| 46 |
+
## Intended uses & limitations
|
| 47 |
+
|
| 48 |
+
More information needed
|
| 49 |
+
|
| 50 |
+
## Training and evaluation data
|
| 51 |
+
|
| 52 |
+
More information needed
|
| 53 |
+
|
| 54 |
+
## Training procedure
|
| 55 |
+
|
| 56 |
+
### Training hyperparameters
|
| 57 |
+
|
| 58 |
+
The following hyperparameters were used during training:
|
| 59 |
+
- learning_rate: 5e-05
|
| 60 |
+
- train_batch_size: 8
|
| 61 |
+
- eval_batch_size: 8
|
| 62 |
+
- seed: 42
|
| 63 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
| 64 |
+
- lr_scheduler_type: cosine
|
| 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 | 2.4496 | 0.0083 | 0.0289 | 0.0024 | 0.0047 | 0.01 | 0.0114 | 0.019 | 0.0813 | 0.1191 | 0.0915 | 0.1324 | 0.1166 | 0.0124 | 0.136 | 0.0099 | 0.1278 | 0.0067 | 0.1192 | 0.0 | 0.0 | 0.0124 | 0.2124 |
|
| 72 |
+
| No log | 2.0 | 214 | 2.2684 | 0.0299 | 0.0782 | 0.0173 | 0.0074 | 0.0189 | 0.0401 | 0.0639 | 0.1661 | 0.1995 | 0.0976 | 0.1604 | 0.2428 | 0.0954 | 0.391 | 0.0145 | 0.1772 | 0.0077 | 0.1871 | 0.0 | 0.0 | 0.0322 | 0.2422 |
|
| 73 |
+
| No log | 3.0 | 321 | 2.0298 | 0.0535 | 0.1159 | 0.0446 | 0.0097 | 0.0477 | 0.0618 | 0.0981 | 0.2257 | 0.268 | 0.133 | 0.2193 | 0.3214 | 0.1778 | 0.5667 | 0.0256 | 0.2215 | 0.0099 | 0.2138 | 0.0008 | 0.0323 | 0.0533 | 0.3058 |
|
| 74 |
+
| No log | 4.0 | 428 | 1.8945 | 0.0859 | 0.1853 | 0.0702 | 0.0334 | 0.0778 | 0.0961 | 0.1189 | 0.293 | 0.3341 | 0.1919 | 0.2761 | 0.4265 | 0.2585 | 0.6189 | 0.049 | 0.3177 | 0.0166 | 0.2567 | 0.0041 | 0.1538 | 0.1011 | 0.3231 |
|
| 75 |
+
| 3.4087 | 5.0 | 535 | 1.8465 | 0.0983 | 0.2227 | 0.0737 | 0.031 | 0.0773 | 0.1063 | 0.1557 | 0.308 | 0.3463 | 0.1614 | 0.2826 | 0.486 | 0.3287 | 0.6045 | 0.0574 | 0.3468 | 0.036 | 0.2786 | 0.0086 | 0.2369 | 0.061 | 0.2644 |
|
| 76 |
+
| 3.4087 | 6.0 | 642 | 1.7933 | 0.1135 | 0.2609 | 0.0872 | 0.029 | 0.0843 | 0.1564 | 0.1418 | 0.3147 | 0.3511 | 0.1587 | 0.2775 | 0.5219 | 0.382 | 0.6018 | 0.0429 | 0.3215 | 0.0267 | 0.2728 | 0.011 | 0.2169 | 0.1048 | 0.3427 |
|
| 77 |
+
| 3.4087 | 7.0 | 749 | 1.7358 | 0.128 | 0.2893 | 0.1046 | 0.0681 | 0.1009 | 0.1695 | 0.1657 | 0.3336 | 0.371 | 0.202 | 0.3124 | 0.5316 | 0.3881 | 0.5892 | 0.0586 | 0.3747 | 0.0395 | 0.3089 | 0.0332 | 0.2569 | 0.1203 | 0.3253 |
|
| 78 |
+
| 3.4087 | 8.0 | 856 | 1.6967 | 0.1554 | 0.346 | 0.1184 | 0.0488 | 0.1298 | 0.2155 | 0.1863 | 0.3443 | 0.372 | 0.1588 | 0.3188 | 0.5313 | 0.4219 | 0.5959 | 0.0906 | 0.3658 | 0.0653 | 0.3058 | 0.0395 | 0.2569 | 0.1595 | 0.3356 |
|
| 79 |
+
| 3.4087 | 9.0 | 963 | 1.6399 | 0.1636 | 0.3504 | 0.1231 | 0.0682 | 0.139 | 0.2298 | 0.1889 | 0.3603 | 0.3918 | 0.2191 | 0.3224 | 0.5498 | 0.418 | 0.5883 | 0.0996 | 0.4038 | 0.0647 | 0.3232 | 0.0303 | 0.2708 | 0.2056 | 0.3729 |
|
| 80 |
+
| 1.5778 | 10.0 | 1070 | 1.5569 | 0.1838 | 0.3943 | 0.143 | 0.0643 | 0.1485 | 0.2628 | 0.2179 | 0.3785 | 0.3994 | 0.2143 | 0.3385 | 0.5573 | 0.4564 | 0.6185 | 0.1166 | 0.3899 | 0.091 | 0.3165 | 0.0554 | 0.3108 | 0.1994 | 0.3613 |
|
| 81 |
+
| 1.5778 | 11.0 | 1177 | 1.5197 | 0.1939 | 0.4116 | 0.1664 | 0.081 | 0.1464 | 0.2871 | 0.2228 | 0.3794 | 0.4032 | 0.1872 | 0.3522 | 0.5597 | 0.4738 | 0.6203 | 0.1204 | 0.4089 | 0.0977 | 0.3058 | 0.0806 | 0.3231 | 0.197 | 0.3582 |
|
| 82 |
+
| 1.5778 | 12.0 | 1284 | 1.4805 | 0.2164 | 0.466 | 0.1818 | 0.0903 | 0.1561 | 0.3301 | 0.2355 | 0.3812 | 0.4044 | 0.1933 | 0.3425 | 0.5688 | 0.4865 | 0.6279 | 0.1528 | 0.4418 | 0.122 | 0.321 | 0.0801 | 0.28 | 0.2405 | 0.3511 |
|
| 83 |
+
| 1.5778 | 13.0 | 1391 | 1.4775 | 0.216 | 0.4679 | 0.1811 | 0.096 | 0.1596 | 0.3276 | 0.2327 | 0.3889 | 0.4121 | 0.2207 | 0.3467 | 0.5831 | 0.4643 | 0.6054 | 0.1596 | 0.438 | 0.1273 | 0.3299 | 0.0984 | 0.3231 | 0.2303 | 0.364 |
|
| 84 |
+
| 1.5778 | 14.0 | 1498 | 1.4550 | 0.2185 | 0.4864 | 0.16 | 0.0813 | 0.1483 | 0.3481 | 0.2396 | 0.3863 | 0.409 | 0.2272 | 0.3341 | 0.5899 | 0.4974 | 0.6428 | 0.1488 | 0.4203 | 0.143 | 0.325 | 0.0954 | 0.2985 | 0.2082 | 0.3587 |
|
| 85 |
+
| 1.3087 | 15.0 | 1605 | 1.4381 | 0.2388 | 0.4949 | 0.2116 | 0.0844 | 0.184 | 0.3583 | 0.2521 | 0.3961 | 0.4194 | 0.1964 | 0.3474 | 0.5867 | 0.5148 | 0.6432 | 0.1774 | 0.4228 | 0.1375 | 0.3152 | 0.1241 | 0.3385 | 0.2401 | 0.3773 |
|
| 86 |
+
| 1.3087 | 16.0 | 1712 | 1.4094 | 0.2482 | 0.5241 | 0.2015 | 0.0931 | 0.1922 | 0.3754 | 0.2535 | 0.4039 | 0.4216 | 0.2291 | 0.3709 | 0.576 | 0.5054 | 0.6419 | 0.1876 | 0.4114 | 0.1622 | 0.3393 | 0.1283 | 0.3338 | 0.2578 | 0.3818 |
|
| 87 |
+
| 1.3087 | 17.0 | 1819 | 1.4044 | 0.2528 | 0.5311 | 0.2088 | 0.096 | 0.2052 | 0.3609 | 0.2595 | 0.4056 | 0.4228 | 0.223 | 0.3795 | 0.5587 | 0.516 | 0.6396 | 0.2009 | 0.4177 | 0.1449 | 0.3384 | 0.143 | 0.3354 | 0.2593 | 0.3831 |
|
| 88 |
+
| 1.3087 | 18.0 | 1926 | 1.3968 | 0.2581 | 0.5282 | 0.2107 | 0.101 | 0.2083 | 0.3787 | 0.2659 | 0.4174 | 0.4363 | 0.2289 | 0.3908 | 0.5821 | 0.5132 | 0.6387 | 0.213 | 0.4797 | 0.1584 | 0.3371 | 0.1375 | 0.3431 | 0.2682 | 0.3827 |
|
| 89 |
+
| 1.1363 | 19.0 | 2033 | 1.3735 | 0.2592 | 0.5349 | 0.2235 | 0.0869 | 0.2076 | 0.3936 | 0.2661 | 0.4127 | 0.4335 | 0.2159 | 0.3829 | 0.5907 | 0.5222 | 0.6414 | 0.2115 | 0.4468 | 0.1608 | 0.3549 | 0.1408 | 0.3431 | 0.2609 | 0.3813 |
|
| 90 |
+
| 1.1363 | 20.0 | 2140 | 1.3686 | 0.266 | 0.5447 | 0.2204 | 0.0897 | 0.2138 | 0.3875 | 0.268 | 0.4117 | 0.4326 | 0.2102 | 0.3875 | 0.5781 | 0.5323 | 0.6446 | 0.1933 | 0.4418 | 0.1727 | 0.3558 | 0.1537 | 0.3246 | 0.278 | 0.396 |
|
| 91 |
+
| 1.1363 | 21.0 | 2247 | 1.3672 | 0.2659 | 0.5446 | 0.2266 | 0.0859 | 0.2198 | 0.3884 | 0.2687 | 0.4164 | 0.4339 | 0.2135 | 0.3914 | 0.5779 | 0.5341 | 0.6505 | 0.1966 | 0.419 | 0.1624 | 0.3513 | 0.1698 | 0.3554 | 0.2668 | 0.3933 |
|
| 92 |
+
| 1.1363 | 22.0 | 2354 | 1.3658 | 0.2731 | 0.5472 | 0.2303 | 0.095 | 0.2174 | 0.4119 | 0.27 | 0.4179 | 0.4383 | 0.202 | 0.3904 | 0.5994 | 0.5311 | 0.6473 | 0.2131 | 0.4494 | 0.179 | 0.3545 | 0.1672 | 0.3492 | 0.2754 | 0.3911 |
|
| 93 |
+
| 1.1363 | 23.0 | 2461 | 1.3644 | 0.2692 | 0.5458 | 0.228 | 0.0937 | 0.2188 | 0.4008 | 0.2681 | 0.4205 | 0.4394 | 0.2104 | 0.3876 | 0.6031 | 0.534 | 0.6518 | 0.2028 | 0.4418 | 0.175 | 0.3594 | 0.1635 | 0.3477 | 0.2708 | 0.3964 |
|
| 94 |
+
| 1.0307 | 24.0 | 2568 | 1.3586 | 0.269 | 0.5419 | 0.2327 | 0.0933 | 0.2161 | 0.4067 | 0.2699 | 0.4208 | 0.439 | 0.2168 | 0.392 | 0.5983 | 0.5384 | 0.6527 | 0.2045 | 0.443 | 0.1728 | 0.3536 | 0.1581 | 0.3508 | 0.2709 | 0.3951 |
|
| 95 |
+
| 1.0307 | 25.0 | 2675 | 1.3503 | 0.2788 | 0.5573 | 0.239 | 0.0992 | 0.2301 | 0.4157 | 0.273 | 0.4225 | 0.4416 | 0.2261 | 0.388 | 0.6083 | 0.5361 | 0.6581 | 0.2253 | 0.4519 | 0.1805 | 0.3634 | 0.1742 | 0.3338 | 0.2779 | 0.4009 |
|
| 96 |
+
| 1.0307 | 26.0 | 2782 | 1.3441 | 0.2803 | 0.5605 | 0.2394 | 0.0922 | 0.2291 | 0.4206 | 0.2702 | 0.4213 | 0.4406 | 0.2174 | 0.387 | 0.6034 | 0.5392 | 0.6604 | 0.2319 | 0.4481 | 0.1831 | 0.3634 | 0.1761 | 0.3369 | 0.2711 | 0.3942 |
|
| 97 |
+
| 1.0307 | 27.0 | 2889 | 1.3461 | 0.2811 | 0.5603 | 0.2465 | 0.0939 | 0.23 | 0.4185 | 0.2741 | 0.4231 | 0.4418 | 0.2217 | 0.3879 | 0.6041 | 0.537 | 0.6545 | 0.2329 | 0.4506 | 0.181 | 0.3638 | 0.1792 | 0.3385 | 0.2753 | 0.4013 |
|
| 98 |
+
| 1.0307 | 28.0 | 2996 | 1.3462 | 0.2799 | 0.5611 | 0.2378 | 0.0943 | 0.231 | 0.4151 | 0.2732 | 0.4211 | 0.4392 | 0.2194 | 0.3863 | 0.5988 | 0.5376 | 0.655 | 0.2291 | 0.4456 | 0.179 | 0.3616 | 0.176 | 0.3323 | 0.2775 | 0.4013 |
|
| 99 |
+
| 0.9602 | 29.0 | 3103 | 1.3458 | 0.2811 | 0.5599 | 0.2393 | 0.094 | 0.2311 | 0.4197 | 0.2734 | 0.4208 | 0.4385 | 0.2191 | 0.3862 | 0.5986 | 0.5393 | 0.6559 | 0.2282 | 0.4405 | 0.1785 | 0.3594 | 0.1788 | 0.3338 | 0.2806 | 0.4031 |
|
| 100 |
+
| 0.9602 | 30.0 | 3210 | 1.3461 | 0.2811 | 0.561 | 0.239 | 0.0945 | 0.2317 | 0.419 | 0.2736 | 0.4208 | 0.4388 | 0.2191 | 0.3871 | 0.598 | 0.5394 | 0.6554 | 0.2284 | 0.4405 | 0.1791 | 0.3598 | 0.1786 | 0.3354 | 0.2802 | 0.4027 |
|
| 101 |
+
|
| 102 |
+
|
| 103 |
+
### Framework versions
|
| 104 |
+
|
| 105 |
+
- Transformers 4.41.2
|
| 106 |
+
- Pytorch 2.3.0+cu121
|
| 107 |
+
- Datasets 2.20.0
|
| 108 |
+
- Tokenizers 0.19.1
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 174079796
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7fb81c3a1092e5b74f015654a6304e5b353d6938036005c76fbed154e4c2519d
|
| 3 |
size 174079796
|