Instructions to use danelcsb/rtdetr_v2_r18vd with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use danelcsb/rtdetr_v2_r18vd with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("object-detection", model="danelcsb/rtdetr_v2_r18vd")# Load model directly from transformers import AutoTokenizer, AutoModelForObjectDetection tokenizer = AutoTokenizer.from_pretrained("danelcsb/rtdetr_v2_r18vd") model = AutoModelForObjectDetection.from_pretrained("danelcsb/rtdetr_v2_r18vd") - Notebooks
- Google Colab
- Kaggle
Add config from convert_rt_detr_v2_original_pytorch_checkpoint_to_pytorch.py
Browse files- config.json +1 -0
config.json
CHANGED
|
@@ -47,6 +47,7 @@
|
|
| 47 |
256
|
| 48 |
],
|
| 49 |
"decoder_layers": 3,
|
|
|
|
| 50 |
"decoder_n_points": 4,
|
| 51 |
"decoder_offset_scale": 0.5,
|
| 52 |
"disable_custom_kernels": true,
|
|
|
|
| 47 |
256
|
| 48 |
],
|
| 49 |
"decoder_layers": 3,
|
| 50 |
+
"decoder_n_levels": 3,
|
| 51 |
"decoder_n_points": 4,
|
| 52 |
"decoder_offset_scale": 0.5,
|
| 53 |
"disable_custom_kernels": true,
|