MCPcopy Create free account

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

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

Method__getitem__
Return one normalized image tensor with box and keypoint targets.
tests/training/test_module_data.py:98
Method__getitem__
(self, idx: int)
tests/training/test_module_data.py:316
Method__getitem__
(self, idx: int)
tests/training/test_module_data.py:372
Method__getitem__
(self, idx)
tests/training/helpers.py:92
Method__getitem__
Return fake series by column name.
tests/visualize/test_training.py:50
Method__getitem__
(self, idx)
tests/datasets/test_save_grids.py:26
Method__getitem__
(self, idx)
tests/datasets/test_augmentations.py:1689
Method__init__
(self, message: str)
src/rfdetr/__init__.py:106
Method__init__
(self, **kwargs)
src/rfdetr/variants.py:174
Method__init__
(self, **kwargs)
src/rfdetr/detr.py:236
Method__init__
( self, model: torch.nn.Module, postprocess: PostProcess, device: torch.device
src/rfdetr/inference.py:39
Method__init__
(self, name: str, level: int = logging.NOTSET)
src/rfdetr/utilities/logger.py:16
Method__init__
(self, tensors: Tensor, mask: Tensor | None)
src/rfdetr/utilities/tensors.py:65
Method__init__
( self, dataset: torch.utils.data.Dataset, effective_batch_size: int, world_si
src/rfdetr/training/module_data.py:65
Method__init__
(self, model_config: ModelConfig, train_config: TrainConfig)
src/rfdetr/training/module_data.py:148
Method__init__
( self, model: torch.nn.Module, decay: float = 0.9997, tau: float = 0,
src/rfdetr/training/model_ema.py:19
Method__init__
(self, init_res: float = 0.0, better: str = "large")
src/rfdetr/training/model_ema.py:66
Method__init__
(self, init_res: float = 0.0, better: str = "large", use_ema: bool = False)
src/rfdetr/training/model_ema.py:103
Method__init__
(self, model_config: ModelConfig, train_config: TrainConfig)
src/rfdetr/training/module_model.py:50
Method__init__
( self, decay: float = 0.993, tau: int = 100, use_buffers: bool = True,
src/rfdetr/training/callbacks/ema.py:36
Method__init__
( self, max_dets: int = DEFAULT_KEYPOINT_MAX_DETS, segmentation: bool = False,
src/rfdetr/training/callbacks/coco_eval.py:108
Method__init__
( self, patience: int = 10, min_delta: float = 0.001, use_ema: bool = False,
src/rfdetr/training/callbacks/best_model.py:595
Method__init__
( self, drop_path: float = 0.0, dropout: float = 0.0, cutoff_epoch: int = 0,
src/rfdetr/training/callbacks/drop_schedule.py:34
Method__init__
( self, mean: tuple[float, ...] = (0.485, 0.456, 0.406), std: tuple[float, ...] = (0.2
src/rfdetr/datasets/transforms.py:43
Method__init__
(self, transform: alb.BasicTransform, keypoint_flip_pairs: list[int] | None = None)
src/rfdetr/datasets/transforms.py:402
Method__init__
(self, classes: list[str], samples: list[_LazyYoloSample])
src/rfdetr/datasets/yolo.py:148
Method__init__
( self, img_folder: str, lb_folder: str, data_file: str, transforms=No
src/rfdetr/datasets/yolo.py:742
Method__init__
(self, num_samples: int = 10, transforms: Any | None = None)
src/rfdetr/datasets/_develop.py:130
Method__init__
( self, img_folder: str | Path, ann_file: str | Path, transforms: Any | None,
src/rfdetr/datasets/coco.py:238
Method__init__
( self, data_loader: DataLoader, output_dir: Path, max_batches: int = 3, dataset_type: str = "train"
src/rfdetr/datasets/save_grids.py:35
Method__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
src/rfdetr/models/math.py:72
Method__init__
Creates the matcher. Args: cost_class: Relative weight of the classification error in the matching cost. cost_bbox: R
src/rfdetr/models/matcher.py:50
Method__init__
Create the criterion. Parameters: num_classes: number of object categories, omitting the special no-object category m
src/rfdetr/models/criterion.py:147
Method__init__
( self, d_model: int = 512, sa_nhead: int = 8, ca_nhead: int = 8, num_
src/rfdetr/models/transformer.py:148
Method__init__
( self, decoder_layer: "TransformerDecoderLayer", num_layers: int, norm: nn.Mo
src/rfdetr/models/transformer.py:516
Method__init__
( self, d_model: int, sa_nhead: int, ca_nhead: int, dim_feedforward: i
src/rfdetr/models/transformer.py:755
Method__init__
Initializes the model. Parameters: backbone: torch module of the backbone to be used. See backbone.py transformer: to
src/rfdetr/models/lwdetr.py:119
Method__init__
( self, num_select: int = 300, num_keypoints_per_class: list[int] | None = None,
src/rfdetr/models/postprocess.py:27
Method__init__
(self, num_pos_feats=256)
src/rfdetr/models/position_encoding.py:102
Method__init__
Multi-Scale Deformable Attention Module :param d_model hidden dimension :param n_levels number of feature levels :param n_heads
src/rfdetr/models/ops/modules/ms_deform_attn.py:38
Method__init__
( self, shape=(640, 640), out_feature_indexes=[2, 4, 5, 9], size="base",
src/rfdetr/models/backbone/dinov2.py:56
Method__init__
(self)
src/rfdetr/models/backbone/base.py:14
Method__init__
(self, backbone, position_embedding)
src/rfdetr/models/backbone/__init__.py:21
Method__init__
(self, normalized_shape, eps=1e-6)
src/rfdetr/models/backbone/projector.py:29
Method__init__
( self, in_planes, out_planes, kernel=3, stride=1, groups=1,
src/rfdetr/models/backbone/projector.py:85
Method__init__
ch_in, ch_out, shortcut, groups, kernels, expand.
src/rfdetr/models/backbone/projector.py:126
Method__init__
Args: in_channels: Channels in each input feature map level. out_channels: Number of channels in the output feature m
src/rfdetr/models/backbone/projector.py:168
Method__init__
(self, in_dim, out_dim, factor_kernel=False)
src/rfdetr/models/backbone/projector.py:299
Method__init__
( self, name: str, pretrained_encoder: str = None, window_block_indexes: list
src/rfdetr/models/backbone/backbone.py:34
Method__init__
( self, hidden_size=768, num_hidden_layers=12, num_attention_heads=12,
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:226
Method__init__
(self, config)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:291
Method__init__
(self, config: WindowedDinov2WithRegistersConfig)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:320
Method__init__
(self, config: WindowedDinov2WithRegistersConfig)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:466
Method__init__
(self, config: WindowedDinov2WithRegistersConfig)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:526
Method__init__
(self, config: WindowedDinov2WithRegistersConfig)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:570
Method__init__
(self, config: WindowedDinov2WithRegistersConfig)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:583
Method__init__
(self, config: WindowedDinov2WithRegistersConfig)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:621
Method__init__
(self, config)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:627
Method__init__
(self, drop_prob: Optional[float] = None)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:657
Method__init__
(self, config)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:688
Method__init__
(self, config: WindowedDinov2WithRegistersConfig)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:713
Method__init__
(self, config: WindowedDinov2WithRegistersConfig)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:784
Method__init__
(self, config: WindowedDinov2WithRegistersConfig)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:912
Method__init__
(self, config: WindowedDinov2WithRegistersConfig)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:1070
Method__init__
(self, config: WindowedDinov2WithRegistersConfig)
src/rfdetr/models/backbone/dinov2_with_windowed_attn.py:1187
Method__init__
(self, hidden_dim: int, num_classes: int)
src/rfdetr/models/heads/detection.py:24
Method__init__
Create the initializer. Args: dim: Query conditioning dimensionality. num_queries: Number of query tokens to instanti
src/rfdetr/models/heads/keypoints.py:70
Method__init__
(self, dim, layer_scale_init_value=0)
src/rfdetr/models/heads/segmentation.py:142
Method__init__
(self, in_dim, num_blocks: int, bottleneck_ratio: int = 1, downsample_ratio: int = 4)
src/rfdetr/models/heads/segmentation.py:211
Method__init__
( self, coco_gt: Any, keypoint_oks_sigmas: list[float] | None = None, max_dets
src/rfdetr/evaluation/keypoint_oks.py:122
Method__init__
( self, coco_gt: Any, iou_types: list[str], max_dets: int = 100, keypo
src/rfdetr/evaluation/coco_eval.py:340
Method__init__
( self, engine_path="dino.engine", device="cuda:0", sync_mode: bool = False, max_batch_size=32, verbos
src/rfdetr/export/benchmark.py:179
Method__init__
( self, )
src/rfdetr/export/benchmark.py:333
Method__init__
(self, input, severity=None)
src/rfdetr/export/_onnx/exporter.py:191
Method__init__
(self)
tests/inference/test_predict_eval_mode.py:22
Method__init__
Initialise stub with optional class names, label list, and keypoint flag.
tests/inference/helpers.py:52
Method__init__
(self)
tests/inference/test_optimize_for_inference.py:20
Method__init__
(self, device: torch.device | str = torch.device("cpu"), resolution: int = 28)
tests/inference/test_optimize_for_inference.py:32
Method__init__
(self, captured_device)
tests/inference/test_optimize_for_inference.py:135
Method__init__
(self, captured_device)
tests/inference/test_optimize_for_inference.py:183
Method__init__
Store initialization kwargs for optional inspection.
tests/utilities/test_file_validation.py:25
Method__init__
Initialize the fake response with content and metadata.
tests/utilities/test_file_validation.py:48
Method__init__
(self, returncode: int)
tests/utilities/test_package.py:28
Method__init__
(self, length: int = 100, with_coco: bool = False)
tests/training/test_module_data.py:70
Method__init__
(self)
tests/training/test_auto_batch.py:77
Method__init__
(self)
tests/training/test_module_model.py:156
Method__init__
(self)
tests/training/helpers.py:28
Method__init__
(self, length: int = 20)
tests/training/helpers.py:71
Method__init__
(self, eval_interval: int = 2)
tests/training/callbacks/test_best_model_callback.py:96
Method__init__
(self)
tests/training/callbacks/test_best_model_callback.py:118
Method__init__
(self)
tests/training/callbacks/test_ema_callback.py:150
Method__init__
(self)
tests/training/callbacks/test_coco_eval_callback.py:491
Method__init__
Store values for ``notna().any()`` checks.
tests/visualize/test_training.py:29
Method__init__
Store column data for ``_build_metric_groups``.
tests/visualize/test_training.py:45
Method__init__
(self, num_samples: int = 4)
tests/datasets/test_save_grids.py:20
Method__init__
(self, *, min_max_height, size, p=1.0)
tests/datasets/test_augmentations.py:29
Method__init__
(self, *, min_max_height, height, width, p=1.0)
tests/datasets/test_augmentations.py:38
Method__init__
(self, transforms)
tests/datasets/test_augmentations.py:1683
Method__init__
(self, augmentation_backend="cpu", square_resize_div_64=False)
tests/datasets/test_coco.py:420
Method__init__
(self, hw: list[tuple[int, int]])
tests/models/test_transformer.py:473
← previousnext →801–900 of 3,167, ranked by callers