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