Instructions to use devmab/rt_detrv2_finetuned_trashify_box_detector_v1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use devmab/rt_detrv2_finetuned_trashify_box_detector_v1 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("object-detection", model="devmab/rt_detrv2_finetuned_trashify_box_detector_v1")# Load model directly from transformers import AutoTokenizer, AutoModelForObjectDetection tokenizer = AutoTokenizer.from_pretrained("devmab/rt_detrv2_finetuned_trashify_box_detector_v1") model = AutoModelForObjectDetection.from_pretrained("devmab/rt_detrv2_finetuned_trashify_box_detector_v1", device_map="auto") - Notebooks
- Google Colab
- Kaggle
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: 8.7848
- Map: 0.5341
- Map 50: 0.7349
- Map 75: 0.6468
- Map Small: 0.0002
- Map Medium: 0.3539
- Map Large: 0.5642
- Mar 1: 0.5616
- Mar 10: 0.6998
- Mar 100: 0.7109
- Mar Small: 0.1
- Mar Medium: 0.4727
- Mar Large: 0.7541
- Map Bin: 0.7906
- Mar 100 Bin: 0.8624
- Map Hand: 0.6251
- Mar 100 Hand: 0.7716
- Map Not Bin: 0.1563
- Mar 100 Not Bin: 0.5571
- Map Not Hand: -1.0
- Mar 100 Not Hand: -1.0
- Map Not Trash: 0.2793
- Mar 100 Not Trash: 0.55
- Map Trash: 0.6638
- Mar 100 Trash: 0.7575
- Map Trash Arm: 0.6895
- Mar 100 Trash Arm: 0.7667
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: 8
- eval_batch_size: 8
- seed: 42
- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- lr_scheduler_warmup_steps: 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 |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 51.5156 | 1.0 | 99 | 13.7205 | 0.2811 | 0.4686 | 0.3044 | 0.075 | 0.1591 | 0.2982 | 0.3627 | 0.5246 | 0.5594 | 0.15 | 0.2886 | 0.6078 | 0.5287 | 0.7908 | 0.4659 | 0.7284 | 0.0577 | 0.4071 | -1.0 | -1.0 | 0.1447 | 0.3806 | 0.481 | 0.6159 | 0.0086 | 0.4333 |
| 27.4746 | 2.0 | 198 | 11.1932 | 0.3675 | 0.6013 | 0.3811 | 0.05 | 0.1226 | 0.386 | 0.4229 | 0.6108 | 0.6767 | 0.1 | 0.3926 | 0.7023 | 0.624 | 0.839 | 0.5015 | 0.7716 | 0.0216 | 0.5786 | -1.0 | -1.0 | 0.1862 | 0.5278 | 0.5246 | 0.7097 | 0.3471 | 0.6333 |
| 21.3245 | 3.0 | 297 | 10.3638 | 0.463 | 0.6557 | 0.5348 | 0.0333 | 0.1794 | 0.4835 | 0.5118 | 0.6528 | 0.6875 | 0.1 | 0.4227 | 0.7176 | 0.7066 | 0.8298 | 0.5809 | 0.7637 | 0.0447 | 0.5143 | -1.0 | -1.0 | 0.2462 | 0.5389 | 0.5823 | 0.7451 | 0.6176 | 0.7333 |
| 17.5271 | 4.0 | 396 | 8.9923 | 0.5053 | 0.6952 | 0.5614 | 0.0029 | 0.2914 | 0.5287 | 0.5283 | 0.6975 | 0.7317 | 0.05 | 0.442 | 0.7685 | 0.789 | 0.8709 | 0.6364 | 0.798 | 0.1239 | 0.6357 | -1.0 | -1.0 | 0.269 | 0.5528 | 0.615 | 0.7327 | 0.5985 | 0.8 |
| 15.4906 | 5.0 | 495 | 8.7578 | 0.5218 | 0.7101 | 0.6176 | 0.0004 | 0.3216 | 0.5469 | 0.5495 | 0.6853 | 0.7186 | 0.15 | 0.4631 | 0.7594 | 0.7915 | 0.8702 | 0.6385 | 0.7873 | 0.1072 | 0.5571 | -1.0 | -1.0 | 0.2808 | 0.5708 | 0.638 | 0.7593 | 0.6751 | 0.7667 |
| 14.0321 | 6.0 | 594 | 8.7554 | 0.5468 | 0.7347 | 0.6472 | 0.0667 | 0.2817 | 0.5789 | 0.5696 | 0.7179 | 0.7412 | 0.2 | 0.5023 | 0.7726 | 0.7938 | 0.8716 | 0.6207 | 0.7912 | 0.2396 | 0.6429 | -1.0 | -1.0 | 0.2668 | 0.5819 | 0.6523 | 0.7593 | 0.7075 | 0.8 |
| 12.7781 | 7.0 | 693 | 8.6576 | 0.5227 | 0.732 | 0.5797 | 0.0375 | 0.3851 | 0.5539 | 0.5719 | 0.704 | 0.729 | 0.15 | 0.55 | 0.7631 | 0.8019 | 0.8702 | 0.6183 | 0.7902 | 0.2485 | 0.6143 | -1.0 | -1.0 | 0.2735 | 0.5833 | 0.6391 | 0.7496 | 0.5552 | 0.7667 |
| 11.7420 | 8.0 | 792 | 8.7110 | 0.529 | 0.7011 | 0.615 | 0.0 | 0.3364 | 0.5583 | 0.5561 | 0.6941 | 0.7159 | 0.0 | 0.454 | 0.7578 | 0.7953 | 0.8567 | 0.6101 | 0.7735 | 0.154 | 0.55 | -1.0 | -1.0 | 0.2872 | 0.5569 | 0.6595 | 0.7584 | 0.6676 | 0.8 |
| 10.8350 | 9.0 | 891 | 8.7799 | 0.5247 | 0.7201 | 0.5654 | 0.0001 | 0.3469 | 0.5567 | 0.5629 | 0.6917 | 0.7066 | 0.1 | 0.4688 | 0.7508 | 0.8018 | 0.8631 | 0.6179 | 0.7804 | 0.1644 | 0.5571 | -1.0 | -1.0 | 0.2831 | 0.55 | 0.6556 | 0.7558 | 0.6255 | 0.7333 |
| 10.2010 | 10.0 | 990 | 8.7848 | 0.5341 | 0.7349 | 0.6468 | 0.0002 | 0.3539 | 0.5642 | 0.5616 | 0.6998 | 0.7109 | 0.1 | 0.4727 | 0.7541 | 0.7906 | 0.8624 | 0.6251 | 0.7716 | 0.1563 | 0.5571 | -1.0 | -1.0 | 0.2793 | 0.55 | 0.6638 | 0.7575 | 0.6895 | 0.7667 |
Framework versions
- Transformers 5.0.0
- Pytorch 2.10.0+cu128
- Datasets 4.8.5
- Tokenizers 0.22.2
- Downloads last month
- 9
Model tree for devmab/rt_detrv2_finetuned_trashify_box_detector_v1
Base model
PekingU/rtdetr_v2_r50vd