Hugging Face's logo

mrdbourke
/
rt_detrv2_finetuned_trashify_box_detector_v1

Object Detection
Transformers
Safetensors
rt_detr_v2
Generated from Trainer
Model card Files
xet
Community
mrdbourke commited on
Commit
46003b6
·
verified ·
1 Parent(s): baf78e0

upload fine-tuned RT-DETRv2 trashify object detection model

Browse files
Files changed (4) hide show
  1. README.md +36 -36
  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: 8.9000
20
- - Map: 0.5134
21
- - Map 50: 0.6917
22
- - Map 75: 0.5749
23
- - Map Small: 0.4
24
- - Map Medium: 0.2845
25
- - Map Large: 0.5538
26
- - Mar 1: 0.5482
27
- - Mar 10: 0.7189
28
- - Mar 100: 0.7663
29
- - Mar Small: 0.4
30
- - Mar Medium: 0.533
31
- - Mar Large: 0.7931
32
- - Map Bin: 0.7876
33
- - Mar 100 Bin: 0.8879
34
- - Map Hand: 0.5723
35
- - Mar 100 Hand: 0.8118
36
- - Map Not Bin: 0.1797
37
- - Mar 100 Not Bin: 0.6857
38
  - Map Not Hand: -1.0
39
  - Mar 100 Not Hand: -1.0
40
- - Map Not Trash: 0.2679
41
- - Mar 100 Not Trash: 0.625
42
- - Map Trash: 0.6726
43
- - Mar 100 Trash: 0.7876
44
- - Map Trash Arm: 0.6
45
- - Mar 100 Trash Arm: 0.8
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
- | 75.2499 | 1.0 | 50 | 17.5113 | 0.2036 | 0.293 | 0.2137 | 0.0 | 0.0349 | 0.2153 | 0.2926 | 0.4248 | 0.508 | 0.0 | 0.1244 | 0.5579 | 0.5792 | 0.8312 | 0.2434 | 0.7696 | 0.0044 | 0.3429 | -1.0 | -1.0 | 0.0107 | 0.4639 | 0.3837 | 0.6407 | 0.0 | 0.0 |
79
- | 23.852 | 2.0 | 100 | 11.4502 | 0.2711 | 0.3799 | 0.3015 | 0.05 | 0.1059 | 0.2818 | 0.3735 | 0.5918 | 0.6483 | 0.35 | 0.3608 | 0.6945 | 0.6972 | 0.9035 | 0.2595 | 0.8088 | 0.0109 | 0.5643 | -1.0 | -1.0 | 0.031 | 0.5958 | 0.6088 | 0.7504 | 0.0192 | 0.2667 |
80
- | 18.2873 | 3.0 | 150 | 10.0729 | 0.4112 | 0.5678 | 0.4869 | 0.3655 | 0.2303 | 0.432 | 0.4785 | 0.6951 | 0.7657 | 0.45 | 0.4551 | 0.7968 | 0.7569 | 0.905 | 0.3534 | 0.8343 | 0.0278 | 0.6571 | -1.0 | -1.0 | 0.1497 | 0.6236 | 0.6421 | 0.7743 | 0.5371 | 0.8 |
81
- | 15.8982 | 4.0 | 200 | 9.4929 | 0.48 | 0.6555 | 0.5578 | 0.4 | 0.2552 | 0.5051 | 0.524 | 0.7099 | 0.7588 | 0.4 | 0.4597 | 0.7931 | 0.753 | 0.8936 | 0.5989 | 0.8353 | 0.1333 | 0.6429 | -1.0 | -1.0 | 0.1993 | 0.6319 | 0.6537 | 0.7823 | 0.542 | 0.7667 |
82
- | 14.6758 | 5.0 | 250 | 9.4786 | 0.47 | 0.6472 | 0.5411 | 0.4 | 0.2494 | 0.5009 | 0.5346 | 0.6907 | 0.7252 | 0.4 | 0.3784 | 0.7732 | 0.7641 | 0.8766 | 0.5657 | 0.8029 | 0.1636 | 0.5571 | -1.0 | -1.0 | 0.2588 | 0.6083 | 0.6364 | 0.7726 | 0.4312 | 0.7333 |
83
- | 13.5443 | 6.0 | 300 | 9.2135 | 0.495 | 0.6699 | 0.5594 | 0.35 | 0.347 | 0.5225 | 0.5432 | 0.7086 | 0.7602 | 0.35 | 0.5625 | 0.7905 | 0.7808 | 0.895 | 0.5788 | 0.8157 | 0.1336 | 0.6286 | -1.0 | -1.0 | 0.2336 | 0.6208 | 0.6626 | 0.8009 | 0.5804 | 0.8 |
84
- | 12.828 | 7.0 | 350 | 8.9653 | 0.5041 | 0.6851 | 0.5799 | 0.35 | 0.2242 | 0.5328 | 0.543 | 0.7152 | 0.7596 | 0.35 | 0.5034 | 0.7952 | 0.7919 | 0.8922 | 0.5883 | 0.8127 | 0.1407 | 0.6643 | -1.0 | -1.0 | 0.2459 | 0.6264 | 0.6884 | 0.7956 | 0.5692 | 0.7667 |
85
- | 12.1564 | 8.0 | 400 | 8.8797 | 0.509 | 0.683 | 0.5708 | 0.35 | 0.2002 | 0.542 | 0.5565 | 0.7412 | 0.7722 | 0.35 | 0.5267 | 0.8006 | 0.782 | 0.8879 | 0.6009 | 0.8137 | 0.1517 | 0.6857 | -1.0 | -1.0 | 0.2626 | 0.6278 | 0.6564 | 0.785 | 0.6003 | 0.8333 |
86
- | 11.5731 | 9.0 | 450 | 9.0043 | 0.5126 | 0.692 | 0.5879 | 0.4 | 0.2861 | 0.5548 | 0.5454 | 0.7211 | 0.7714 | 0.4 | 0.5199 | 0.8015 | 0.7828 | 0.8823 | 0.5674 | 0.8176 | 0.2052 | 0.6929 | -1.0 | -1.0 | 0.2661 | 0.6139 | 0.6843 | 0.7885 | 0.5698 | 0.8333 |
87
- | 11.2251 | 10.0 | 500 | 8.9000 | 0.5134 | 0.6917 | 0.5749 | 0.4 | 0.2845 | 0.5538 | 0.5482 | 0.7189 | 0.7663 | 0.4 | 0.533 | 0.7931 | 0.7876 | 0.8879 | 0.5723 | 0.8118 | 0.1797 | 0.6857 | -1.0 | -1.0 | 0.2679 | 0.625 | 0.6726 | 0.7876 | 0.6 | 0.8 |
88
 
89
 
90
  ### Framework versions
91
 
92
- - Transformers 4.52.3
93
  - Pytorch 2.7.0+cu126
94
  - Datasets 3.6.0
95
  - Tokenizers 0.21.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: 10.5656
20
+ - Map: 0.464
21
+ - Map 50: 0.6416
22
+ - Map 75: 0.5044
23
+ - Map Small: 0.0536
24
+ - Map Medium: 0.0797
25
+ - Map Large: 0.4902
26
+ - Mar 1: 0.5143
27
+ - Mar 10: 0.6661
28
+ - Mar 100: 0.7398
29
+ - Mar Small: 0.35
30
+ - Mar Medium: 0.4801
31
+ - Mar Large: 0.7745
32
+ - Map Bin: 0.6777
33
+ - Mar 100 Bin: 0.8596
34
+ - Map Hand: 0.4888
35
+ - Mar 100 Hand: 0.7716
36
+ - Map Not Bin: 0.1536
37
+ - Mar 100 Not Bin: 0.6286
38
  - Map Not Hand: -1.0
39
  - Mar 100 Not Hand: -1.0
40
+ - Map Not Trash: 0.2042
41
+ - Mar 100 Not Trash: 0.6056
42
+ - Map Trash: 0.3921
43
+ - Mar 100 Trash: 0.7071
44
+ - Map Trash Arm: 0.8673
45
+ - Mar 100 Trash Arm: 0.8667
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
+ | 187.7449 | 1.0 | 50 | 76.7837 | 0.0758 | 0.142 | 0.0703 | 0.0 | 0.0145 | 0.0786 | 0.1331 | 0.2875 | 0.3285 | 0.0 | 0.1858 | 0.3657 | 0.1694 | 0.644 | 0.1982 | 0.35 | 0.0103 | 0.4286 | -1.0 | -1.0 | 0.002 | 0.1708 | 0.075 | 0.3779 | 0.0 | 0.0 |
79
+ | 62.3477 | 2.0 | 100 | 24.7528 | 0.1999 | 0.3065 | 0.186 | 0.0 | 0.056 | 0.2109 | 0.3267 | 0.5336 | 0.5503 | 0.0 | 0.1756 | 0.5765 | 0.2835 | 0.7525 | 0.4412 | 0.6922 | 0.0064 | 0.3357 | -1.0 | -1.0 | 0.1136 | 0.3042 | 0.1756 | 0.6841 | 0.1789 | 0.5333 |
80
+ | 25.8089 | 3.0 | 150 | 14.7906 | 0.2551 | 0.3877 | 0.276 | 0.0 | 0.0733 | 0.2665 | 0.3814 | 0.5903 | 0.6508 | 0.0 | 0.3176 | 0.6945 | 0.2075 | 0.866 | 0.461 | 0.6892 | 0.0172 | 0.5071 | -1.0 | -1.0 | 0.0982 | 0.4111 | 0.1641 | 0.6646 | 0.5827 | 0.7667 |
81
+ | 18.0845 | 4.0 | 200 | 11.7929 | 0.3316 | 0.4912 | 0.3748 | 0.05 | 0.0913 | 0.3544 | 0.4116 | 0.6086 | 0.688 | 0.1 | 0.5091 | 0.7201 | 0.4974 | 0.8674 | 0.5367 | 0.7853 | 0.0997 | 0.6286 | -1.0 | -1.0 | 0.1815 | 0.4653 | 0.2816 | 0.6814 | 0.3929 | 0.7 |
82
+ | 15.2751 | 5.0 | 250 | 11.5743 | 0.3691 | 0.5764 | 0.3935 | 0.15 | 0.0582 | 0.3909 | 0.4324 | 0.5942 | 0.6722 | 0.15 | 0.3432 | 0.7168 | 0.5633 | 0.8518 | 0.5121 | 0.7951 | 0.0899 | 0.5857 | -1.0 | -1.0 | 0.1651 | 0.5153 | 0.3176 | 0.6522 | 0.5665 | 0.6333 |
83
+ | 13.6233 | 6.0 | 300 | 10.9435 | 0.424 | 0.6113 | 0.4833 | 0.0889 | 0.0624 | 0.4464 | 0.5064 | 0.6589 | 0.731 | 0.3 | 0.3568 | 0.7686 | 0.6258 | 0.8617 | 0.5329 | 0.7676 | 0.1067 | 0.6143 | -1.0 | -1.0 | 0.1775 | 0.5889 | 0.3769 | 0.6867 | 0.7243 | 0.8667 |
84
+ | 12.4361 | 7.0 | 350 | 10.4613 | 0.4421 | 0.6414 | 0.509 | 0.0667 | 0.0751 | 0.4698 | 0.4951 | 0.6401 | 0.7209 | 0.2 | 0.3869 | 0.7596 | 0.6413 | 0.8915 | 0.4859 | 0.7676 | 0.1556 | 0.6214 | -1.0 | -1.0 | 0.221 | 0.5819 | 0.4148 | 0.6965 | 0.7338 | 0.7667 |
85
+ | 11.6579 | 8.0 | 400 | 10.5329 | 0.4304 | 0.6103 | 0.4732 | 0.0354 | 0.0619 | 0.4564 | 0.4889 | 0.6422 | 0.7183 | 0.35 | 0.2438 | 0.7635 | 0.6411 | 0.8582 | 0.5113 | 0.7725 | 0.1577 | 0.4857 | -1.0 | -1.0 | 0.1848 | 0.5889 | 0.3841 | 0.7044 | 0.7031 | 0.9 |
86
+ | 11.0091 | 9.0 | 450 | 10.6710 | 0.4553 | 0.6325 | 0.5138 | 0.0667 | 0.0794 | 0.4824 | 0.5038 | 0.6475 | 0.7273 | 0.4 | 0.3267 | 0.7714 | 0.6649 | 0.8638 | 0.48 | 0.7618 | 0.1481 | 0.5571 | -1.0 | -1.0 | 0.2132 | 0.5986 | 0.4013 | 0.7159 | 0.8243 | 0.8667 |
87
+ | 10.5976 | 10.0 | 500 | 10.5656 | 0.464 | 0.6416 | 0.5044 | 0.0536 | 0.0797 | 0.4902 | 0.5143 | 0.6661 | 0.7398 | 0.35 | 0.4801 | 0.7745 | 0.6777 | 0.8596 | 0.4888 | 0.7716 | 0.1536 | 0.6286 | -1.0 | -1.0 | 0.2042 | 0.6056 | 0.3921 | 0.7071 | 0.8673 | 0.8667 |
88
 
89
 
90
  ### Framework versions
91
 
92
+ - Transformers 4.52.4
93
  - Pytorch 2.7.0+cu126
94
  - Datasets 3.6.0
95
  - Tokenizers 0.21.1
config.json CHANGED
@@ -125,7 +125,7 @@
125
  "num_queries": 300,
126
  "positional_encoding_temperature": 10000,
127
  "torch_dtype": "float32",
128
- "transformers_version": "4.52.3",
129
  "use_focal_loss": true,
130
  "use_pretrained_backbone": false,
131
  "use_timm_backbone": false,
 
125
  "num_queries": 300,
126
  "positional_encoding_temperature": 10000,
127
  "torch_dtype": "float32",
128
+ "transformers_version": "4.52.4",
129
  "use_focal_loss": true,
130
  "use_pretrained_backbone": false,
131
  "use_timm_backbone": false,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7419f7006fb3d491453145609d63546a304c17354697d137408d23242ffabe52
3
  size 171576780
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0ae530c0805b2068827d10fab1a274e9471cafaa3796d94394cce138a3349d6
3
  size 171576780
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3782ebdc15a8999842a98f0e26e37cb1bb7b65e6e2adba862d5d55f94a6079c5
3
  size 5777
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9f763dea8cbe4baac252ff1b8525f520877517ae483fa1f1e79e61335a62c2e
3
  size 5777