MCPcopy Create free account

hub / github.com/roboflow/rf-detr / functions

Functions3,167 in github.com/roboflow/rf-detr

Method__init__
(self, hidden_dim: int, num_keypoints_per_class: list[int])
tests/models/test_lwdetr_keypoints.py:43
Method__init__
(self, transformer: Transformer)
tests/models/test_transformer_onnx_spatial_shapes.py:94
Method__init__
(self)
tests/models/backbone/test_backbone_export.py:37
Method__init__
( self, features: list[NestedTensor], cross_attention_features: list[object] | None,
tests/models/backbone/test_backbone.py:20
Method__init__
(self)
tests/export/test_tflite_export.py:54
Method__init__
(self, *, segmentation_head: bool = False)
tests/export/test_export.py:52
Method__len__
Return the padded dataset length (always a multiple of the alignment unit).
src/rfdetr/training/module_data.py:95
Method__len__
(self)
src/rfdetr/datasets/yolo.py:152
Method__len__
(self)
src/rfdetr/datasets/yolo.py:791
Method__len__
(self)
src/rfdetr/datasets/_develop.py:134
Method__len__
(self)
tests/training/test_module_data.py:79
Method__len__
Return the fixed fake dataset length.
tests/training/test_module_data.py:94
Method__len__
(self)
tests/training/test_module_data.py:313
Method__len__
(self)
tests/training/test_module_data.py:369
Method__len__
(self)
tests/training/helpers.py:74
Method__len__
(self)
tests/datasets/test_save_grids.py:23
Method__len__
(self)
tests/datasets/test_augmentations.py:1686
Method__post_init__
Validate that ratios sum to approximately 1.0 and are non-negative.
src/rfdetr/datasets/synthetic.py:45
Method__repr__
(self)
src/rfdetr/utilities/tensors.py:106
Method__repr__
(self)
src/rfdetr/training/model_ema.py:92
Method__repr__
Return a readable string representation of the wrapper. Returns: Representation including the wrapped transform and type.
src/rfdetr/datasets/transforms.py:436
Method__repr__
(self)
tests/utilities/test_tensors.py:63
Method__str__
(self)
src/rfdetr/training/model_ema.py:135
Method_all_gather
(x: list)
tests/evaluation/test_coco_eval.py:378
Function_arange_with_device_capture
(*args, **kwargs)
tests/utilities/test_box_ops.py:41
Method_build
(image_set, args, resolution)
tests/training/test_module_data.py:408
Method_build
(image_set, args, resolution)
tests/training/test_module_data.py:1055
Method_build
(image_set, args, resolution)
tests/training/test_module_data.py:1432
Function_callback
()
tests/training/test_train_resume.py:66
Method_capture
(msg: str, *args: object, **kwargs: object)
tests/models/test_weights.py:921
Method_capture
(msg, *args, **kwargs)
tests/models/test_weights.py:987
Method_capture_init
(self_t, **kwargs)
tests/models/test_config.py:436
Method_capture_update
(preds, targets)
tests/training/callbacks/test_coco_eval_callback.py:245
Method_capture_warning
(msg, *args, **kwargs)
tests/models/test_weights.py:1141
Method_coerce_amp_dtype
Fall back to ``'auto'`` (with a warning) for an unrecognised or wrong-typed ``amp_dtype``. Mixed precision is a best-effort speed/memory opti
src/rfdetr/config.py:768
Method_coerce_legacy_progress_bar
Normalize legacy boolean progress_bar values to the new string/None representation. This preserves compatibility with older configs where ``p
src/rfdetr/config.py:757
Method_coerce_resume_path
Normalise the resume checkpoint value to ``str`` without resolving it. Unlike ``dataset_dir``/``output_dir``, ``resume`` is forwarded verbati
src/rfdetr/config.py:875
Function_collate
(batch)
tests/datasets/test_save_grids.py:37
Method_compute_with_terminal_summary
()
tests/training/callbacks/test_coco_eval_callback.py:956
Method_counting_setup
()
tests/training/test_module_data.py:1459
Method_counting_training_step
(batch, batch_idx)
tests/training/test_trainer_smoke.py:115
Method_counting_val_step
(batch, batch_idx)
tests/training/test_trainer_smoke.py:152
Function_ensure_model_on_device
Decorate RF-DETR instance methods that require lazy model device placement. The wrapped method receives the same arguments and return value as th
src/rfdetr/detr.py:209
Function_fake_all_gather
(output_tensors, input_tensor)
tests/utilities/test_distributed.py:16
Method_fake_build_kornia
(aug_cfg, resolution, with_masks=False)
tests/training/test_module_data.py:1137
Function_fake_check_output
(command: list[str], cwd: str | None = None)
tests/utilities/test_package.py:20
Function_fake_export_onnx
(*_args, dynamic_axes=None, **_kw)
tests/export/test_export.py:180
Method_fake_export_onnx
(*_a, **_kw)
tests/export/test_export.py:504
Method_fake_export_onnx
(*_args, variant_name=None, **_kw)
tests/export/test_export.py:745
Method_fake_find_spec
(fullname: str, path: list[str] | None = None, target: object | None = None)
tests/training/test_detr_shim.py:1129
Function_fake_make_infer_image
(*_args, **_kwargs)
tests/export/test_export.py:177
Method_fake_make_infer_image
(*_a, **_kw)
tests/export/test_export.py:501
Method_fake_make_infer_image
(*_args, **_kwargs)
tests/export/test_export.py:742
Function_fake_onnx_export
(*args, **kwargs)
tests/export/test_export.py:81
Method_fake_onnx_export
(*args, **kwargs)
tests/export/test_export.py:644
Function_fake_probe
(*args, **kwargs)
tests/training/test_auto_batch.py:33
Function_fake_run
(command, shell, capture_output, text, check)
tests/export/test_tensorrt_export.py:17
Method_fake_trainer
(**kwargs)
tests/training/test_build_trainer.py:368
Method_fake_trainer
(**kwargs)
tests/training/test_build_trainer.py:514
Method_fake_trainer
(**kwargs)
tests/training/test_build_trainer.py:618
Method_fake_trainer
(**kwargs)
tests/training/test_build_trainer.py:825
Method_fake_trainer
(**kwargs)
tests/training/test_build_trainer.py:907
Method_fake_trainer
(**kwargs)
tests/training/test_build_trainer.py:1019
Method_fake_trainer
(**kwargs)
tests/training/test_build_trainer.py:1048
Method_flip_idx_to_pairs
Convert Ultralytics ``flip_idx`` permutation metadata to flat swap pairs.
src/rfdetr/detr.py:1564
Method_get_coco_api
(dataset)
tests/training/callbacks/test_coco_eval_callback.py:776
Function_get_tensor
(index: int)
tests/export/test_tflite_inference.py:74
Method_get_tensor
(index: int)
tests/export/test_tflite_inference.py:491
Method_get_tensor
(index: int)
tests/export/test_tflite_inference.py:588
Method_get_tgt_permutation_idx
(self, indices)
src/rfdetr/models/criterion.py:586
Method_import_with_missing_ipython
(name: str, *args, **kwargs)
tests/training/callbacks/test_coco_eval_callback.py:84
Method_init_weights
Initialize the weights.
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:848
Function_make
(**overrides)
tests/training/conftest.py:26
Function_make_affine
Build a ``K.RandomAffine`` from aug_config params. Albumentations ``translate_percent`` is a ``(min, max)`` signed range (e.g. ``(-0.1, 0.1)``).
src/rfdetr/datasets/kornia_transforms.py:177
Function_make_color_jitter
Build a ``K.ColorJiggle`` from aug_config ``ColorJitter`` params. Note: Kornia >=0.7 uses ``ColorJiggle``; the ``ColorJitter`` alias was adde
src/rfdetr/datasets/kornia_transforms.py:205
Function_make_detr_args
Return a SimpleNamespace shaped like the args passed to _load_pretrain_weights_into.
tests/training/test_detr_shim.py:1332
Function_make_gauss_noise
Build a ``K.RandomGaussianNoise`` from aug_config params. Kornia takes a single ``std`` value; we use the upper bound of ``std_range`` as an acce
src/rfdetr/datasets/kornia_transforms.py:252
Function_make_gaussian_blur
Build a ``K.RandomGaussianBlur`` from aug_config params. ``blur_limit`` is rounded up to an odd number for the kernel size.
src/rfdetr/datasets/kornia_transforms.py:233
Function_make_horizontal_flip
Build a ``K.RandomHorizontalFlip`` from aug_config params.
src/rfdetr/datasets/kornia_transforms.py:142
Method_make_ns
(self, mc=None, tc=None)
tests/models/test_namespace.py:85
Function_make_random_brightness_contrast
Build a ``K.ColorJiggle`` from ``RandomBrightnessContrast`` params.
src/rfdetr/datasets/kornia_transforms.py:222
Function_make_rotate
Build a ``K.RandomRotation`` from aug_config params. The ``limit`` parameter may be a scalar (symmetric range) or a tuple.
src/rfdetr/datasets/kornia_transforms.py:156
Function_make_train_config
Return a minimal TrainConfig for use in load_pretrain_weights. Args: tmp_path: Optional pytest tmp_path fixture value.
tests/models/test_weights.py:50
Function_make_vertical_flip
Build a ``K.RandomVerticalFlip`` from aug_config params.
src/rfdetr/datasets/kornia_transforms.py:149
Function_meshgrid_with_indexing_assertion
(*args, **kwargs)
tests/utilities/test_box_ops.py:17
Function_meshgrid_with_indexing_assertion
(*args, **kwargs)
tests/models/test_transformer.py:71
Function_meshgrid_wrong_indexing
(*args, **kwargs)
tests/models/test_transformer.py:115
Method_mock_ai_edge_litert
Inject a fake ai_edge_litert.interpreter into sys.modules and mask lower-priority backends. Mirrors ``_mock_tflite_runtime`` for the first-pr
tests/export/test_tflite_inference.py:216
Method_mock_import
(name, *args, **kwargs)
tests/training/test_module_data.py:1077
Method_mock_import
(name, globals=None, locals=None, fromlist=(), level=0)
tests/training/test_detr_shim.py:177
Method_mock_import
(name, *args, **kwargs)
tests/datasets/test_coco.py:532
Method_mock_onnx_info
Mock ``_get_onnx_input_info`` to return a known shape.
tests/export/test_tflite_export.py:768
Method_mock_tflite_runtime
Inject a fake tflite_runtime.interpreter into sys.modules and mask ai_edge_litert. ``_create_interpreter`` probes backends in priority order:
tests/export/test_tflite_inference.py:114
Function_move_side_effect
(model: object)
tests/training/test_auto_batch.py:161
Method_mutating_export
()
tests/inference/test_optimize_for_inference.py:512
Method_normalize_device
Normalize supported device inputs to a canonical torch-style string. Args: v: Device specifier provided by callers. Supported val
src/rfdetr/config.py:395
Function_onnx_runtime
Benchmark ONNX Runtime inference for one image and provider list. Creates a fresh ``InferenceSession`` with the requested providers, preprocesses
src/rfdetr/export/_onnx/inference.py:253
Method_patch_download
Suppress all download and file-existence side effects.
tests/training/test_detr_shim.py:1378
Method_patch_download
Suppress all download and file-existence side effects.
tests/training/test_load_pretrain_weights.py:121
Method_patch_download
Suppress all download and file-existence side effects.
tests/training/test_load_pretrain_weights.py:270
← previousnext →901–1,000 of 3,167, ranked by callers