Hugging Face's logo

Sairii
/
rt_detrv2_finetuned_trashify_box_detector_v1

Object Detection
Transformers
Safetensors
rt_detr_v2
Generated from Trainer
Model card Files
xet
 Community
Sairii commited on
Commit
d857dff
·
verified ·
1 Parent(s): 25d8065

upload finetuned RT-DETRv2 model on trashify dataset from learn HF course

Browse files
Files changed (4) hide show
  1. README.md +38 -38
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. training_args.bin +1 -1
README.md CHANGED
@@ -16,33 +16,33 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [PekingU/rtdetr_v2_r50vd](https://huggingface.co/PekingU/rtdetr_v2_r50vd) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 9.3604
20
- - Map: 0.4734
21
- - Map 50: 0.6189
22
- - Map 75: 0.5492
23
- - Map Small: 0.0043
24
- - Map Medium: 0.3139
25
- - Map Large: 0.4859
26
- - Mar 1: 0.5342
27
- - Mar 10: 0.6855
28
- - Mar 100: 0.7314
29
- - Mar Small: 0.6
30
- - Mar Medium: 0.4369
31
- - Mar Large: 0.7579
32
- - Map Bin: 0.7824
33
- - Mar 100 Bin: 0.8917
34
- - Map Hand: 0.6187
35
- - Mar 100 Hand: 0.8394
36
- - Map Not Bin: 0.1687
37
- - Mar 100 Not Bin: 0.5727
38
- - Map Not Hand: 0.0152
39
- - Mar 100 Not Hand: 0.4667
40
- - Map Not Trash: 0.2546
41
- - Mar 100 Not Trash: 0.6206
42
- - Map Trash: 0.6453
43
- - Mar 100 Trash: 0.8289
44
- - Map Trash Arm: 0.829
45
- - Mar 100 Trash Arm: 0.9
46
 
47
  ## Model description
48
 
@@ -75,21 +75,21 @@ The following hyperparameters were used during training:
75
 
76
  | 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 Bin | Mar 100 Bin | Map Hand | Mar 100 Hand | Map Not Bin | Mar 100 Not Bin | Map Not Hand | Mar 100 Not Hand | Map Not Trash | Mar 100 Not Trash | Map Trash | Mar 100 Trash | Map Trash Arm | Mar 100 Trash Arm |
77
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:----------:|:---------:|:------:|:------:|:-------:|:---------:|:----------:|:---------:|:-------:|:-----------:|:--------:|:------------:|:-----------:|:---------------:|:------------:|:----------------:|:-------------:|:-----------------:|:---------:|:-------------:|:-------------:|:-----------------:|
78
- | 66.9088 | 1.0 | 99 | 12.5494 | 0.295 | 0.4225 | 0.3243 | 0.0 | 0.1243 | 0.304 | 0.3218 | 0.5333 | 0.6105 | 0.0 | 0.417 | 0.6525 | 0.6494 | 0.8667 | 0.5287 | 0.8039 | 0.0085 | 0.6571 | -1.0 | -1.0 | 0.0659 | 0.5486 | 0.5176 | 0.7867 | 0.0 | 0.0 |
79
- | 19.6198 | 2.0 | 198 | 10.2937 | 0.4212 | 0.6279 | 0.4303 | 0.2 | 0.2526 | 0.4393 | 0.4972 | 0.6584 | 0.7064 | 0.2 | 0.5727 | 0.7316 | 0.7621 | 0.8787 | 0.4822 | 0.8186 | 0.0454 | 0.5857 | -1.0 | -1.0 | 0.1808 | 0.5653 | 0.6042 | 0.7566 | 0.4527 | 0.6333 |
80
- | 16.7153 | 3.0 | 297 | 9.7917 | 0.4607 | 0.6309 | 0.5465 | 0.0678 | 0.2368 | 0.4763 | 0.5462 | 0.7051 | 0.7628 | 0.35 | 0.6415 | 0.784 | 0.7618 | 0.8979 | 0.5816 | 0.8461 | 0.0496 | 0.6643 | -1.0 | -1.0 | 0.2329 | 0.6417 | 0.605 | 0.7938 | 0.5334 | 0.7333 |
81
- | 14.9577 | 4.0 | 396 | 9.5954 | 0.492 | 0.6749 | 0.5735 | 0.0833 | 0.2403 | 0.5155 | 0.5694 | 0.7022 | 0.7585 | 0.2 | 0.5773 | 0.7832 | 0.7629 | 0.9064 | 0.5959 | 0.8431 | 0.1525 | 0.6071 | -1.0 | -1.0 | 0.2492 | 0.6389 | 0.6332 | 0.7885 | 0.5582 | 0.7667 |
82
- | 13.625 | 5.0 | 495 | 9.4667 | 0.5116 | 0.6828 | 0.5934 | 0.25 | 0.2328 | 0.5436 | 0.578 | 0.727 | 0.7658 | 0.25 | 0.5523 | 0.7963 | 0.7687 | 0.8972 | 0.6278 | 0.8441 | 0.2472 | 0.5786 | -1.0 | -1.0 | 0.2168 | 0.6139 | 0.5975 | 0.7947 | 0.6113 | 0.8667 |
83
- | 12.5412 | 6.0 | 594 | 9.3681 | 0.5514 | 0.7376 | 0.638 | 0.15 | 0.2464 | 0.5809 | 0.5813 | 0.7235 | 0.7807 | 0.3 | 0.5426 | 0.8113 | 0.772 | 0.9028 | 0.6247 | 0.8382 | 0.2631 | 0.6214 | -1.0 | -1.0 | 0.2537 | 0.6153 | 0.6454 | 0.8062 | 0.7498 | 0.9 |
84
- | 11.6236 | 7.0 | 693 | 9.5192 | 0.5276 | 0.7016 | 0.609 | 0.0113 | 0.2142 | 0.5587 | 0.5813 | 0.7078 | 0.7559 | 0.35 | 0.542 | 0.7879 | 0.7603 | 0.8957 | 0.6307 | 0.8304 | 0.203 | 0.5786 | -1.0 | -1.0 | 0.2584 | 0.5764 | 0.6343 | 0.7876 | 0.6791 | 0.8667 |
85
- | 10.7813 | 8.0 | 792 | 9.5629 | 0.5366 | 0.699 | 0.6114 | 0.0071 | 0.2387 | 0.5691 | 0.5903 | 0.7345 | 0.7623 | 0.3 | 0.571 | 0.7918 | 0.7649 | 0.8929 | 0.5928 | 0.8186 | 0.216 | 0.5929 | -1.0 | -1.0 | 0.2631 | 0.5569 | 0.632 | 0.8124 | 0.7507 | 0.9 |
86
- | 10.0775 | 9.0 | 891 | 9.6664 | 0.5185 | 0.6783 | 0.5917 | 0.0029 | 0.2411 | 0.5495 | 0.5928 | 0.7203 | 0.7679 | 0.3 | 0.5551 | 0.8037 | 0.7604 | 0.8929 | 0.5898 | 0.8127 | 0.1925 | 0.6357 | -1.0 | -1.0 | 0.2487 | 0.5681 | 0.6444 | 0.7982 | 0.675 | 0.9 |
87
- | 9.5537 | 10.0 | 990 | 9.7227 | 0.5307 | 0.6908 | 0.608 | 0.0015 | 0.2338 | 0.5654 | 0.5868 | 0.7269 | 0.7598 | 0.3 | 0.5756 | 0.7894 | 0.7511 | 0.8936 | 0.6076 | 0.8176 | 0.1917 | 0.6 | -1.0 | -1.0 | 0.244 | 0.5458 | 0.6393 | 0.8018 | 0.7507 | 0.9 |
88
 
89
 
90
  ### Framework versions
91
 
92
- - Transformers 4.57.1
93
  - Pytorch 2.9.0+cu126
94
  - Datasets 4.4.1
95
  - Tokenizers 0.22.1
 
16
 
17
  This model is a fine-tuned version of [PekingU/rtdetr_v2_r50vd](https://huggingface.co/PekingU/rtdetr_v2_r50vd) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 9.3653
20
+ - Map: 0.4704
21
+ - Map 50: 0.618
22
+ - Map 75: 0.5266
23
+ - Map Small: 0.1026
24
+ - Map Medium: 0.2908
25
+ - Map Large: 0.4828
26
+ - Mar 1: 0.5158
27
+ - Mar 10: 0.6877
28
+ - Mar 100: 0.7583
29
+ - Mar Small: 0.3
30
+ - Mar Medium: 0.5958
31
+ - Mar Large: 0.7711
32
+ - Map Bin: 0.7558
33
+ - Mar 100 Bin: 0.8896
34
+ - Map Hand: 0.5647
35
+ - Mar 100 Hand: 0.8303
36
+ - Map Not Bin: 0.2042
37
+ - Mar 100 Not Bin: 0.6455
38
+ - Map Not Hand: 0.0105
39
+ - Mar 100 Not Hand: 0.5667
40
+ - Map Not Trash: 0.2291
41
+ - Mar 100 Not Trash: 0.6246
42
+ - Map Trash: 0.66
43
+ - Mar 100 Trash: 0.8088
44
+ - Map Trash Arm: 0.8684
45
+ - Mar 100 Trash Arm: 0.9429
46
 
47
  ## Model description
48
 
 
75
 
76
  | 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 Bin | Mar 100 Bin | Map Hand | Mar 100 Hand | Map Not Bin | Mar 100 Not Bin | Map Not Hand | Mar 100 Not Hand | Map Not Trash | Mar 100 Not Trash | Map Trash | Mar 100 Trash | Map Trash Arm | Mar 100 Trash Arm |
77
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:----------:|:---------:|:------:|:------:|:-------:|:---------:|:----------:|:---------:|:-------:|:-----------:|:--------:|:------------:|:-----------:|:---------------:|:------------:|:----------------:|:-------------:|:-----------------:|:---------:|:-------------:|:-------------:|:-----------------:|
78
+ | 57.1776 | 1.0 | 99 | 12.1891 | 0.3086 | 0.4438 | 0.3387 | 0.0333 | 0.114 | 0.3246 | 0.3508 | 0.5271 | 0.6823 | 0.1 | 0.4625 | 0.7229 | 0.612 | 0.8652 | 0.5289 | 0.7833 | 0.0463 | 0.5714 | -1.0 | -1.0 | 0.0677 | 0.4986 | 0.5952 | 0.7416 | 0.0018 | 0.6333 |
79
+ | 19.2556 | 2.0 | 198 | 10.2258 | 0.4434 | 0.6141 | 0.4754 | 0.026 | 0.2734 | 0.4646 | 0.5363 | 0.7089 | 0.7726 | 0.2 | 0.575 | 0.7988 | 0.7199 | 0.8915 | 0.5813 | 0.8363 | 0.0478 | 0.7214 | -1.0 | -1.0 | 0.1977 | 0.5944 | 0.6426 | 0.7584 | 0.4707 | 0.8333 |
80
+ | 16.3006 | 3.0 | 297 | 9.6586 | 0.5218 | 0.6948 | 0.5849 | 0.1024 | 0.2487 | 0.547 | 0.5751 | 0.727 | 0.7829 | 0.4 | 0.6142 | 0.8073 | 0.7634 | 0.8908 | 0.6461 | 0.8186 | 0.1558 | 0.6786 | -1.0 | -1.0 | 0.2344 | 0.65 | 0.63 | 0.7929 | 0.7014 | 0.8667 |
81
+ | 14.5834 | 4.0 | 396 | 9.1042 | 0.534 | 0.7172 | 0.5876 | 0.0542 | 0.2993 | 0.5633 | 0.5635 | 0.7389 | 0.774 | 0.4 | 0.6199 | 0.7976 | 0.7635 | 0.8957 | 0.6249 | 0.8196 | 0.2098 | 0.6357 | -1.0 | -1.0 | 0.2607 | 0.6278 | 0.6669 | 0.7982 | 0.6782 | 0.8667 |
82
+ | 13.3038 | 5.0 | 495 | 9.0995 | 0.5344 | 0.7296 | 0.6096 | 0.0675 | 0.3141 | 0.5644 | 0.5934 | 0.7289 | 0.7779 | 0.3 | 0.575 | 0.8045 | 0.7825 | 0.8915 | 0.6116 | 0.8284 | 0.2252 | 0.6643 | -1.0 | -1.0 | 0.3111 | 0.6292 | 0.625 | 0.7876 | 0.6507 | 0.8667 |
83
+ | 12.2002 | 6.0 | 594 | 9.3112 | 0.5342 | 0.7325 | 0.6158 | 0.0639 | 0.2493 | 0.5612 | 0.5668 | 0.7374 | 0.7774 | 0.35 | 0.5585 | 0.8062 | 0.7403 | 0.8972 | 0.6094 | 0.8284 | 0.1976 | 0.6571 | -1.0 | -1.0 | 0.3088 | 0.6194 | 0.6414 | 0.7956 | 0.7076 | 0.8667 |
84
+ | 11.1618 | 7.0 | 693 | 9.2933 | 0.548 | 0.752 | 0.6233 | 0.0194 | 0.2432 | 0.58 | 0.5803 | 0.7302 | 0.786 | 0.35 | 0.4932 | 0.8164 | 0.7404 | 0.8929 | 0.5828 | 0.8235 | 0.2804 | 0.6714 | -1.0 | -1.0 | 0.3071 | 0.6389 | 0.6528 | 0.7894 | 0.7245 | 0.9 |
85
+ | 10.3588 | 8.0 | 792 | 9.4387 | 0.5405 | 0.722 | 0.6118 | 0.0292 | 0.2995 | 0.5651 | 0.5943 | 0.7331 | 0.7757 | 0.35 | 0.4682 | 0.8083 | 0.7325 | 0.8645 | 0.5679 | 0.8118 | 0.2327 | 0.6286 | -1.0 | -1.0 | 0.2657 | 0.6167 | 0.6705 | 0.7991 | 0.7739 | 0.9333 |
86
+ | 9.6829 | 9.0 | 891 | 9.4239 | 0.5626 | 0.7531 | 0.6295 | 0.0061 | 0.2474 | 0.5903 | 0.6113 | 0.7554 | 0.7928 | 0.25 | 0.4926 | 0.8219 | 0.7295 | 0.8695 | 0.5958 | 0.8186 | 0.2768 | 0.6786 | -1.0 | -1.0 | 0.3065 | 0.6028 | 0.6512 | 0.7876 | 0.816 | 1.0 |
87
+ | 9.1381 | 10.0 | 990 | 9.4604 | 0.5564 | 0.7409 | 0.6287 | 0.0018 | 0.2661 | 0.5862 | 0.6061 | 0.7412 | 0.7663 | 0.25 | 0.4511 | 0.8004 | 0.7389 | 0.866 | 0.5777 | 0.801 | 0.2743 | 0.6286 | -1.0 | -1.0 | 0.2984 | 0.5847 | 0.65 | 0.7841 | 0.7991 | 0.9333 |
88
 
89
 
90
  ### Framework versions
91
 
92
+ - Transformers 4.57.2
93
  - Pytorch 2.9.0+cu126
94
  - Datasets 4.4.1
95
  - Tokenizers 0.22.1
config.json CHANGED
@@ -126,7 +126,7 @@
126
  "num_feature_levels": 3,
127
  "num_queries": 300,
128
  "positional_encoding_temperature": 10000,
129
- "transformers_version": "4.57.1",
130
  "use_focal_loss": true,
131
  "use_pretrained_backbone": false,
132
  "use_timm_backbone": false,
 
126
  "num_feature_levels": 3,
127
  "num_queries": 300,
128
  "positional_encoding_temperature": 10000,
129
+ "transformers_version": "4.57.2",
130
  "use_focal_loss": true,
131
  "use_pretrained_backbone": false,
132
  "use_timm_backbone": false,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7716a5c2ba7225f2b1d4b814513d4bcd348d723e6366580a549f4f3f793b3467
3
  size 171576780
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e59be21810dd013b951c411512e487b5c99765a4b1d1c25f7e5e9b44476f8d63
3
  size 171576780
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8709c1e2ae58a9bb2198ec226d0e135f4fc537491a16d897ada9d2914551a9d2
3
  size 5905
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6aa5cb9701ddb4b31d9fed947703353740c7f16a89413ff9309a42e95e9bbda1
3
  size 5905