riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-20 04:58:22 +06:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity
c8c8dffbe4
transformers/tests/models/rt_detr
History
Yoni Gozlan 48872fd6ae
Add Image Processor Fast RT-DETR (#34354) 
* add fast image processor rtdetr

* add gpu/cpu test and fix docstring

* remove prints

* add to doc

* nit docstring

* avoid iterating over images/annotations several times

* change torch typing

* Add image processor fast documentation
2024-10-30 13:49:47 -04:00
..
__init__.py New model support RTDETR (#29077) 2024-06-21 17:50:08 +01:00
test_image_processing_rt_detr.py Add Image Processor Fast RT-DETR (#34354) 2024-10-30 13:49:47 -04:00
test_modeling_rt_detr_resnet.py New model support RTDETR (#29077) 2024-06-21 17:50:08 +01:00
test_modeling_rt_detr.py Fix RT-DETR weights initialization (#31724) 2024-07-03 14:29:02 +01:00