Feat: Add Model RF-DETR

DESCRIPTION

@@ -99,6 +99,7 @@ Collate:
     'models-mobilenetv3.R'
     'models-mobilenetv3_large.R'
     'models-resnet.R'
+    'models-rfdetr_detection.R'
     'models-vgg.R'
     'models-vit.R'
     'ops-box_convert.R'
@@ -115,4 +116,4 @@ Depends:
     R (>= 3.5)
 LazyData: true
 VignetteBuilder: knitr
-Config/roxygen2/version: 8.0.0
+RoxygenNote: 8.0.0

NAMESPACE

@@ -169,6 +169,13 @@ export(model_resnet34)
 export(model_resnet50)
 export(model_resnext101_32x8d)
 export(model_resnext50_32x4d)
+export(model_rfdetr_base)
+export(model_rfdetr_base_2)
+export(model_rfdetr_base_o365)
+export(model_rfdetr_large)
+export(model_rfdetr_medium)
+export(model_rfdetr_nano)
+export(model_rfdetr_small)
 export(model_vgg11)
 export(model_vgg11_bn)
 export(model_vgg13)
@@ -263,38 +270,49 @@ importFrom(torch,nn_batch_norm2d)
 importFrom(torch,nn_conv2d)
 importFrom(torch,nn_conv_transpose2d)
 importFrom(torch,nn_dropout)
+importFrom(torch,nn_embedding)
 importFrom(torch,nn_gelu)
 importFrom(torch,nn_hardsigmoid)
 importFrom(torch,nn_hardswish)
 importFrom(torch,nn_identity)
+importFrom(torch,nn_init_constant_)
+importFrom(torch,nn_init_ones_)
+importFrom(torch,nn_init_trunc_normal_)
+importFrom(torch,nn_init_xavier_uniform_)
+importFrom(torch,nn_init_zeros_)
 importFrom(torch,nn_layer_norm)
 importFrom(torch,nn_linear)
 importFrom(torch,nn_max_pool2d)
 importFrom(torch,nn_module)
 importFrom(torch,nn_module_dict)
 importFrom(torch,nn_module_list)
+importFrom(torch,nn_multihead_attention)
 importFrom(torch,nn_parameter)
 importFrom(torch,nn_prelu)
 importFrom(torch,nn_relu)
 importFrom(torch,nn_sequential)
+importFrom(torch,nn_silu)
 importFrom(torch,nn_softmax)
 importFrom(torch,nnf_gelu)
 importFrom(torch,nnf_grid_sample)
 importFrom(torch,nnf_interpolate)
 importFrom(torch,nnf_layer_norm)
 importFrom(torch,nnf_normalize)
 importFrom(torch,nnf_relu)
+importFrom(torch,nnf_silu)
 importFrom(torch,nnf_softmax)
 importFrom(torch,torch_arange)
 importFrom(torch,torch_cat)
 importFrom(torch,torch_chunk)
 importFrom(torch,torch_clamp)
+importFrom(torch,torch_cumsum)
 importFrom(torch,torch_empty)
 importFrom(torch,torch_flatten)
 importFrom(torch,torch_float)
 importFrom(torch,torch_float32)
 importFrom(torch,torch_int32)
 importFrom(torch,torch_linspace)
+importFrom(torch,torch_log)
 importFrom(torch,torch_log2)
 importFrom(torch,torch_long)
 importFrom(torch,torch_matmul)
@@ -305,6 +323,7 @@ importFrom(torch,torch_ones)
 importFrom(torch,torch_randn)
 importFrom(torch,torch_sigmoid)
 importFrom(torch,torch_stack)
+importFrom(torch,torch_sum)
 importFrom(torch,torch_tensor)
 importFrom(torch,torch_zeros)
 importFrom(torch,torch_zeros_like)

NEWS.md

@@ -3,6 +3,7 @@
 ## Bug fixes and improvements
 
 * `nms()` now uses `torchvisionlib::ops_nms()` when torchvisionlib is installed, speeding up inference for `model_fasterrcnn_*()` and `model_maskrcnn_*()` (#321, #322).
+* Added RF-DETR (Nano, Small, Medium, Base, Base-v2, Base-O365, Large) object detection models with DINOv2 backbone and deformable-attention decoder (@DerrickUnleashed, #327).
 
 # torchvision 0.9.0