MCPcopy Index your code

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

Endpoints92 in github.com/roboflow/rf-detr

⨍ Functions 3,167◇ Types & classes 522↳ Endpoints 92
Routebuild_namespace
deprecated(target=_namespace_from_configs, deprecated_in="1.7.0", remove_in="1.9.0")
src/rfdetr/_namespace.py:None
Routeexpand_paths
field_validator("dataset_dir", "output_dir", mode="before")
src/rfdetr/config.py:None
Routeforward
replace_return_docstrings(output_type=BaseModelOutputWithPooling, config_class=_CONFIG_FOR_DOC)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:None
Routeload_pretrain_weights
deprecated(target=True, args_mapping={"train_config": None}, deprecated_in="1.7.0", remove_in="1.9.0", num_war
src/rfdetr/models/weights.py:None
Routetest_absolute_path_resolves_to_known_model
patch("rfdetr.assets.model_weights._download_file")
tests/assets/test_downloads.py:None
Routetest_active_keypoint_counts_filters_zeros
pytest.mark.parametrize( "counts,expected", [ pytest.param([0, 17, 25], [17, 25], id="leading-
tests/datasets/test_keypoint_schema.py:None
Routetest_adapts_height_width_for_v2_api
mock.patch("rfdetr.datasets.transforms.alb.RandomSizedCrop", new=_FakeRandomSizedCropV2)
tests/datasets/test_augmentations.py:None
Routetest_adapts_size_for_v1_api
mock.patch("rfdetr.datasets.transforms.alb.RandomSizedCrop", new=_FakeRandomSizedCropV1)
tests/datasets/test_augmentations.py:None
Routetest_amp_true_ddp_notebook_probes_bf16_normally
patch("torch.cuda.is_available", return_value=True)
tests/training/test_build_trainer.py:None
Routetest_auto_batch_related_fields_reject_non_positive_values
pytest.mark.parametrize( "field,value", [ ("batch_size", 0), ("grad_ac
tests/models/test_config.py:None
Routetest_calls_lora_config_with_correct_target_modules
patch("peft.get_peft_model")
tests/training/test_module_model.py:None
Routetest_characterization_infers_correct_class_dict
pytest.mark.parametrize( ("pretrain_weights, patch_target"), [ ("rf-detr-small.pth
tests/inference/test_from_checkpoint.py:None
Routetest_characterization_infers_correct_class_namespace
pytest.mark.parametrize( ("pretrain_weights, patch_target"), [ ("rf-detr-nano.pth"
tests/inference/test_from_checkpoint.py:None
Routetest_class_count_match_does_not_reinitialize
patch("rfdetr.models.weights.torch.load")
tests/training/test_module_model.py:None
Routetest_class_count_mismatch_triggers_reinitialize
patch("rfdetr.models.weights.torch.load")
tests/training/test_module_model.py:None
Routetest_clip_gradients_delegates_to_super_when_not_bf16
patch("rfdetr.training.module_model.torch.cuda.is_bf16_supported", return_value=True)
tests/training/test_module_model.py:None
Routetest_clip_gradients_uses_clip_grad_norm_when_bf16_mixed
patch("rfdetr.training.module_model.torch.cuda.is_bf16_supported", return_value=True)
tests/training/test_module_model.py:None
Routetest_compatible_checkpoint_does_not_raise
patch("rfdetr.models.weights.torch.load")
tests/training/test_module_model.py:None
Routetest_compile_disabled_when_train_accelerator_is_cpu
patch("rfdetr.training.module_model.torch.compile")
tests/training/test_module_model.py:None
Routetest_concatenates_each_state_across_ranks
patch("rfdetr.training.callbacks.coco_eval.get_world_size", return_value=2)
tests/training/callbacks/test_coco_eval_callback.py:None
Routetest_configure_optimizers_returns_required_key
patch("rfdetr.training.module_model.get_param_dict")
tests/training/test_module_model.py:None
Routetest_cuda_device_context_manager_is_used_for_cuda_device
patch("rfdetr.detr._move_model_context_to_device")
tests/inference/test_optimize_for_inference.py:None
Routetest_cuda_device_context_uses_model_device
patch("rfdetr.detr._move_model_context_to_device")
tests/inference/test_optimize_for_inference.py:None
Routetest_ddp_notebook_raises_clear_error_when_private_launcher_is_missing
patch("rfdetr.training.trainer._InteractiveSpawnLauncher", None)
tests/training/test_build_trainer.py:None
Routetest_detection_checkpoint_into_seg_model_raises
patch("rfdetr.models.weights.torch.load")
tests/training/test_module_model.py:None
Routetest_download_before_load_when_weights_absent
patch("rfdetr.models.weights.os.path.isfile", return_value=False)
tests/training/test_module_model.py:None
Routetest_download_file_http_error
patch("rfdetr.utilities.files.requests.get")
tests/utilities/test_file_validation.py:None
Routetest_download_file_md5_failure_cleans_temp
patch("rfdetr.utilities.files.tqdm", _DummyTqdm)
tests/utilities/test_file_validation.py:None
Routetest_download_file_missing_content_length
patch("rfdetr.utilities.files.tqdm", _DummyTqdm)
tests/utilities/test_file_validation.py:None
Routetest_download_file_replace_failure_cleans_temp
patch("rfdetr.utilities.files.tqdm", _DummyTqdm)
tests/utilities/test_file_validation.py:None
Routetest_download_file_stream_error_cleans_temp
patch("rfdetr.utilities.files.tqdm", _DummyTqdm)
tests/utilities/test_file_validation.py:None
Routetest_download_file_uses_fdopen_without_reopen
patch("rfdetr.utilities.files.tqdm", _DummyTqdm)
tests/utilities/test_file_validation.py:None
Routetest_download_flow_for_real_model
patch("rfdetr.assets.model_weights.os.path.exists")
tests/assets/test_downloads.py:None
Routetest_errors_on_partial_hw_with_v2_api
mock.patch("rfdetr.datasets.transforms.alb.RandomSizedCrop", new=_FakeRandomSizedCropV2)
tests/datasets/test_augmentations.py:None
Routetest_export_bool_patch_size_raises
pytest.mark.parametrize("bad_patch_size", [True, False])
tests/export/test_export.py:None
Routetest_export_invalid_patch_size_raises
pytest.mark.parametrize("bad_patch_size", [0, -1])
tests/export/test_export.py:None
Routetest_falls_back_to_cuda_when_accelerator_module_absent
patch("rfdetr.config.torch")
tests/models/test_config.py:None
Routetest_from_config_partial_height_is_silently_skipped
mock.patch("rfdetr.datasets.transforms.alb.RandomSizedCrop", new=_FakeRandomSizedCropV2)
tests/datasets/test_augmentations.py:None
Routetest_fused_optimizer_disabled_when_cuda_unavailable
patch("rfdetr.training.module_model.torch.cuda.is_available", return_value=False)
tests/training/test_module_model.py:None
Routetest_fused_optimizer_disabled_when_precision_not_bf16
patch("rfdetr.training.module_model.get_param_dict")
tests/training/test_module_model.py:None
Routetest_fused_optimizer_enabled_when_precision_is_bf16_mixed
patch("rfdetr.training.module_model.get_param_dict")
tests/training/test_module_model.py:None
Routetest_gpu_postprocess_flag
pytest.mark.parametrize( "segmentation_head, augmentation_backend, resolved_backend, expected_gpu_post
tests/datasets/test_coco.py:None
Routetest_handles_missing_platform_models_gracefully
patch("rfdetr.assets.model_weights.ModelWeights.from_filename", return_value=None)
tests/assets/test_downloads.py:None
Routetest_handles_missing_rfdetr_plus_gracefully
patch("rfdetr.assets.model_weights.os.path.exists")
tests/assets/test_downloads.py:None
Routetest_infer_keypoint_flip_pairs_from_names_edge_cases
pytest.mark.parametrize( "names,expected", [ pytest.param([], [], id="empty-input"), p
tests/datasets/test_keypoint_schema.py:None
Routetest_legacy_signature_available_accelerator_is_kept
patch("rfdetr.config.torch")
tests/models/test_config.py:None
Routetest_legacy_signature_runtime_error_on_fallback_returns_cpu
patch("rfdetr.config.torch")
tests/models/test_config.py:None
Routetest_legacy_signature_unavailable_accelerator_returns_cpu
patch("rfdetr.config.torch")
tests/models/test_config.py:None
Routetest_link_target
pytest.mark.parametrize( "source, expected_target", [ pytest.param("model.model_co
tests/training/test_cli.py:None
Routetest_loads_checkpoint_successfully
patch("rfdetr.models.weights.torch.load")
tests/training/test_module_model.py:None
Routetest_logs_info_messages
patch("rfdetr.assets.model_weights._validate_file_md5")
tests/assets/test_downloads.py:None
Routetest_logs_warning_on_incorrect_md5
patch("rfdetr.assets.model_weights._download_file")
tests/assets/test_downloads.py:None
Routetest_lr_lambda_cosine_reads_train_config_fields
patch("rfdetr.training.module_model.get_param_dict")
tests/training/test_module_model.py:None
Routetest_lr_lambda_step_decay_after_drop
patch("rfdetr.training.module_model.get_param_dict")
tests/training/test_module_model.py:None
Routetest_lr_lambda_step_decay_before_drop
patch("rfdetr.training.module_model.get_param_dict")
tests/training/test_module_model.py:None
Routetest_lr_lambda_warmup_phase
patch("rfdetr.training.module_model.get_param_dict")
tests/training/test_module_model.py:None
Routetest_model_name_resolves_correct_class
pytest.mark.parametrize( ("model_name, patch_target"), [ ("RFDETRNano", "RFDETRNan
tests/inference/test_from_checkpoint.py:None
Routetest_nested_absolute_path_resolves_to_known_model
patch("rfdetr.assets.model_weights._download_file")
tests/assets/test_downloads.py:None
Routetest_no_op_when_world_size_one
patch("rfdetr.training.callbacks.coco_eval.get_world_size", return_value=1)
tests/training/callbacks/test_coco_eval_callback.py:None
Routetest_nonexistent_model_returns_early
patch("rfdetr.assets.model_weights.ModelWeights.from_filename", return_value=None)
tests/assets/test_downloads.py:None
Routetest_optimizer_is_adamw
patch("rfdetr.training.module_model.get_param_dict")
tests/training/test_module_model.py:None
Routetest_output_names
pytest.mark.parametrize( "segmentation_head, backbone_only, expected_output_names", [
tests/export/test_export.py:None
Routetest_partial_load_is_invoked_during_load_pretrain_weights
patch("rfdetr.models.weights.torch.load")
tests/models/test_weights.py:None
Routetest_patch_size_inferred_from_projection_weight
pytest.mark.parametrize( "ckpt_patch_size,model_patch_size,should_raise", [ pytest
tests/utilities/test_state_dict.py:None
Routetest_patch_size_mismatch_raises
patch("rfdetr.models.weights.torch.load")
tests/training/test_module_model.py:None
Routetest_pe_in_checkpoint_is_interpolated_to_model_resolution
pytest.mark.parametrize( "src_pe_size, tgt_resolution, patch_size, expected_tgt_pe_size", [
tests/training/test_load_pretrain_weights.py:None
Routetest_pin_memory_defaults_to_false_when_accelerator_is_cpu
patch("rfdetr.config.DEVICE", "cuda")
tests/training/test_module_data.py:None
Routetest_predict_invalid_patch_size_raises
pytest.mark.parametrize("bad_patch_size", [0, -1, True, False])
tests/inference/test_predict.py:None
Routetest_projection_inference_silently_skips_when_incomplete
pytest.mark.parametrize( "checkpoint,model_args_kwargs", [ pytest.param(
tests/utilities/test_state_dict.py:None
Routetest_promoted_field_accepts_explicit_value
pytest.mark.parametrize( "field, value", [ pytest.param("clip_max_norm", 0.5, id="
tests/models/test_config.py:None
Routetest_query_embedding_trimmed_to_configured_count
patch("rfdetr.models.weights.torch.load")
tests/training/test_module_model.py:None
Routetest_redownloads_on_load_failure
patch("rfdetr.models.weights.os.path.isfile", return_value=True)
tests/training/test_module_model.py:None
Routetest_replaces_encoder_with_peft_model
patch("peft.get_peft_model")
tests/training/test_module_model.py:None
Routetest_returns_accelerator_when_runtime_available
patch("rfdetr.config.torch")
tests/models/test_config.py:None
Routetest_returns_cpu_when_accelerator_compiled_in_but_unavailable
patch("rfdetr.config.torch")
tests/models/test_config.py:None
Routetest_returns_cpu_when_current_accelerator_raises
patch("rfdetr.config.torch")
tests/models/test_config.py:None
Routetest_returns_cpu_when_no_gpu_available
patch("rfdetr.config.torch")
tests/models/test_config.py:None
Routetest_returns_dict_with_required_key
pytest.mark.parametrize( "key", [ pytest.param("results", id="results-key"),
tests/training/test_module_model.py:None
Routetest_scalar_size_passes_through_on_v1_legacy_path
mock.patch("rfdetr.datasets.transforms.alb.RandomSizedCrop", new=_FakeRandomSizedCropV1)
tests/datasets/test_augmentations.py:None
Routetest_scheduler_interval_is_step
patch("rfdetr.training.module_model.get_param_dict")
tests/training/test_module_model.py:None
Routetest_seed_at_rank_zero
patch("rfdetr.training.module_model.seed_everything")
tests/training/test_module_model.py:None
Routetest_seed_rank_offset
patch("rfdetr.training.module_model.seed_everything")
tests/training/test_module_model.py:None
Routetest_seed_skipped_when_none
patch("rfdetr.training.module_model.seed_everything")
tests/training/test_module_model.py:None
Routetest_seg_checkpoint_into_detection_model_raises
patch("rfdetr.models.weights.torch.load")
tests/training/test_module_model.py:None
Routetest_single_image_rounding_parametrized
pytest.mark.parametrize( "block_size,shape,expected", [ pytest.param(32, (100, 100
tests/utilities/test_tensors.py:None
Routetest_size_takes_precedence_over_hw_on_v2_api
mock.patch("rfdetr.datasets.transforms.alb.RandomSizedCrop", new=_FakeRandomSizedCropV2)
tests/datasets/test_augmentations.py:None
Routetest_total_steps_divided_by_grad_accum_for_keypoint_module
patch("rfdetr.training.module_model.get_param_dict")
tests/training/test_module_model.py:None
Routetest_train_auto_batch_ensures_model_on_device_before_resolve
patch("rfdetr.detr.is_main_process", return_value=False)
tests/training/test_auto_batch.py:None
Routetest_train_dir_fields_accept_path
pytest.mark.parametrize( "field", [ pytest.param("dataset_dir", id="dataset_dir"),
tests/models/test_config.py:None
Routetest_unanimous_gate_runs_when_all_ranks_have_ema
patch("rfdetr.training.callbacks.coco_eval.is_dist_avail_and_initialized", return_value=True)
tests/training/callbacks/test_coco_eval_callback.py:None
Routetest_unanimous_gate_skips_when_a_peer_lacks_ema
patch("rfdetr.training.callbacks.coco_eval.is_dist_avail_and_initialized", return_value=True)
tests/training/callbacks/test_coco_eval_callback.py:None
Routevalidate_positive_train_steps
field_validator( "grad_accum_steps", "auto_batch_target_effective", "auto_batch_max_targets_per_image"
src/rfdetr/config.py:None