rt_detrv2_finetuned_trashify_box_detector_v1
This model is a fine-tuned version of PekingU/rtdetr_v2_r50vd on an unknown dataset. It achieves the following results on the evaluation set:
- Loss: 10.5656
- Map: 0.464
- Map 50: 0.6416
- Map 75: 0.5044
- Map Small: 0.0536
- Map Medium: 0.0797
- Map Large: 0.4902
- Mar 1: 0.5143
- Mar 10: 0.6661
- Mar 100: 0.7398
- Mar Small: 0.35
- Mar Medium: 0.4801
- Mar Large: 0.7745
- Map Bin: 0.6777
- Mar 100 Bin: 0.8596
- Map Hand: 0.4888
- Mar 100 Hand: 0.7716
- Map Not Bin: 0.1536
- Mar 100 Not Bin: 0.6286
- Map Not Hand: -1.0
- Mar 100 Not Hand: -1.0
- Map Not Trash: 0.2042
- Mar 100 Not Trash: 0.6056
- Map Trash: 0.3921
- Mar 100 Trash: 0.7071
- Map Trash Arm: 0.8673
- Mar 100 Trash Arm: 0.8667
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 0.0001
- train_batch_size: 16
- eval_batch_size: 16
- seed: 42
- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- lr_scheduler_warmup_ratio: 0.05
- num_epochs: 10
- mixed_precision_training: Native AMP
Training results
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 |
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
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 |
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 |
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 |
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 |
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 |
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 |
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 |
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 |
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 |
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 |
Framework versions
- Transformers 4.52.4
- Pytorch 2.7.0+cu126
- Datasets 3.6.0
- Tokenizers 0.21.1
- Downloads last month
- 73
Inference Providers
NEW
This model isn't deployed by any Inference Provider.
๐
Ask for provider support
Model tree for mrdbourke/rt_detrv2_finetuned_trashify_box_detector_v1
Base model
PekingU/rtdetr_v2_r50vd