MCPcopy Create free account

hub / github.com/Lans1ng/OSSOD / functions

Functions128 in github.com/Lans1ng/OSSOD

⨍ Functions 128◇ Types & classes 33
↓ 10 callersMethodupdate
(self, num_tasks=1)
mmdet_extension/apis/test.py:40
↓ 9 callersMethodupdate
(self)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:152
↓ 6 callersMethod_bbox_forward
Box head forward function used in both training and testing.
mmdet_extension/models/roi_head/standard_roi_head_mb.py:83
↓ 5 callersMethodcuda
Since ema_model is registered as a plain object, it is necessary to put the ema model to cuda when calling cuda function.
mmdet_extension/models/detectors/semi_base.py:129
↓ 5 callersMethodforward_train
(self, img, img_metas, gt_bboxes,
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:169
↓ 4 callersMethodparse_loss
(losses)
mmdet_extension/models/detectors/semi_base.py:94
↓ 4 callersFunctionsingle_gpu_test
(model, data_loader, show=False, out_dir=None,
mmdet_extension/apis/test.py:62
↓ 3 callersMethod_parse_data_info
(self, idx)
mmdet_extension/datasets/txt_style.py:99
↓ 3 callersFunctionget_palette
Get palette from various inputs. Args: palette (list[tuple] | str | tuple | :obj:`Color`): palette inputs. num_classes (int): the
mmdet_extension/core/visualization/palette.py:22
↓ 3 callersFunctionimshow_det_bboxes
Draw bboxes and class labels (with scores) on an image. Args: img (str | ndarray): The image to be displayed. bboxes (ndarray): B
mmdet_extension/core/visualization/image.py:207
↓ 3 callersFunctionmulti_gpu_test
Test model with multiple gpus. This method tests model with multiple gpus and collects the results under two different modes: gpu and cpu mod
mmdet_extension/apis/test.py:115
↓ 2 callersFunction_get_adaptive_scales
Get adaptive scales according to areas. The scale range is [0.5, 1.0]. When the area is less than ``'min_area'``, the scale is 0.5 while the
mmdet_extension/core/visualization/image.py:38
↓ 2 callersMethodadd
(self, items, class_ids)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:56
↓ 2 callersFunctionbbox2result
(bboxes, labels, num_classes)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:28
↓ 2 callersFunctiondraw_labels
Draw labels on the axes. Args: ax (matplotlib.Axes): The input axes. labels (ndarray): The labels with the shape of (n, ).
mmdet_extension/core/visualization/image.py:112
↓ 2 callersMethodevaluation_once
(self, runner)
mmdet_extension/core/hooks/semi_eval_hooks.py:32
↓ 2 callersMethodget_ann_info
(self, idx)
mmdet_extension/datasets/txt_style.py:103
↓ 2 callersMethodget_params
Get parameters for ``erase`` for a random erasing. Args: img (np.array): ndarray image of size (H, W, C) to be erased.
mmdet_extension/datasets/pipelines/transforms.py:30
↓ 2 callersFunctionpalette_val
Convert palette to matplotlib palette. Args: palette List[tuple]: A list of color tuples. Returns: List[tuple[float]]: A lis
mmdet_extension/core/visualization/palette.py:6
↓ 2 callersMethodsimple_test_bboxes
Test only det bboxes without augmentation.
mmdet_extension/models/roi_head/standard_roi_head_mb.py:96
↓ 2 callersMethodupdate_ema_model
(self, momentum=0.99)
mmdet_extension/models/detectors/semi_base.py:117
↓ 1 callersMethod__init__
(self, dataloader, start=None, interval=1,
mmdet_extension/core/hooks/semi_eval_hooks.py:18
↓ 1 callersMethod__init__
(self, total_num_classes, items_per_class, shuffle=False)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:50
↓ 1 callersMethod__str__
(self)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:80
↓ 1 callersMethod_bbox_forward_train
Run forward function and calculate loss for box head in training.
mmdet_extension/models/roi_head/standard_roi_head_mb.py:68
↓ 1 callersMethod_broadcast_bn_buffer
(self, runner, has_ema)
mmdet_extension/core/hooks/semi_eval_hooks.py:81
↓ 1 callersFunction_get_bias_color
Get different colors for each masks. Get different colors for each masks by adding a bias color to the base category color. Args:
mmdet_extension/core/visualization/image.py:61
↓ 1 callersMethod_parse_str_info
(self, str_data)
mmdet_extension/datasets/txt_style.py:68
↓ 1 callersMethodanalysis
(self)
mmdet_extension/models/detectors/unbiased_teacher.py:140
↓ 1 callersMethodanalysis
(self)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:375
↓ 1 callersFunctioncolormap
Args: rgb (bool): whether to return RGB colors or BGR colors. maximum (int): either 255 or 1 Returns: ndarray: a flo
mmdet_extension/core/utils/colormap.py:88
↓ 1 callersMethodcreate_pseudo_results
using dynamic score to create pseudo results
mmdet_extension/models/detectors/unbiased_teacher.py:109
↓ 1 callersMethodcreate_pseudo_results
using dynamic score to create pseudo results
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:311
↓ 1 callersFunctiondraw_bboxes
Draw bounding boxes on the axes. Args: ax (matplotlib.Axes): The input axes. bboxes (ndarray): The input bounding boxes with the
mmdet_extension/core/visualization/image.py:79
↓ 1 callersFunctiondraw_masks
Draw masks on the image and their edges on the axes. Args: ax (matplotlib.Axes): The input axes. img (ndarray): The image with th
mmdet_extension/core/visualization/image.py:165
↓ 1 callersMethodembeddings_test
Test without augmentation.
mmdet_extension/models/detectors/semi_base.py:61
↓ 1 callersFunctionfeat2result
(feats, labels, num_classes)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:39
↓ 1 callersMethodforward_train_semi
( self, img, img_metas, gt_bboxes, gt_labels, img_unlabeled, img_metas_unlabeled, gt_b
mmdet_extension/models/detectors/unbiased_teacher.py:61
↓ 1 callersMethodinference_unlabeled
(self, img, img_metas, rescale=True, return_feat=False)
mmdet_extension/models/detectors/semi_two_stage.py:33
↓ 1 callersMethodinference_unlabeled
(self, img, img_metas, rescale=True, return_feats=True)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:282
↓ 1 callersMethodload_annotations
(self, ann_file)
mmdet_extension/datasets/txt_style.py:58
↓ 1 callersFunctionmain
()
eval.py:103
↓ 1 callersFunctionmain
()
train.py:90
↓ 1 callersMethodmodel_init
(self, imgs, img_metas, **kwargs)
mmdet_extension/models/detectors/semi_base.py:55
↓ 1 callersFunctionparse_args
()
eval.py:18
↓ 1 callersFunctionparse_args
()
train.py:22
↓ 1 callersMethodretrieve
(self, class_id)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:60
↓ 1 callersMethodset_epoch
(self, epoch)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:147
↓ 1 callersMethodshow_result
(self, img, result, score_thr=0.3, bbox_color=
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:435
↓ 1 callersMethodstart
(self)
mmdet_extension/apis/test.py:33
↓ 1 callersFunctiontrain_detector
(model, dataset, cfg, distributed=False,
mmdet_extension/apis/train.py:22
↓ 1 callersMethodvisual
(self, img, boxes_list, labels_list, img_metas, img_id=[], boxes_ignore_list=None, gt_f
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:384
↓ 1 callersMethodvisual_offline
(self, img, boxes_list, labels_list, img_metas, img_id=[], boxes_ignore_list=None, gt_f
mmdet_extension/models/detectors/unbiased_teacher.py:149
Method__call__
(self, results)
mmdet_extension/datasets/pipelines/transforms.py:65
Method__call__
(self, results)
mmdet_extension/datasets/pipelines/transforms_box.py:18
Method__call__
(self, bboxes, labels, masks=None)
mmdet_extension/datasets/pipelines/transforms_box.py:33
Method__call__
Call function to resize images, bounding boxes, masks, semantic segmentation map. Args: results (dict): Result dict from
mmdet_extension/datasets/pipelines/transforms_box.py:47
Method__call__
(self, bboxes, labels, masks=None)
mmdet_extension/datasets/pipelines/transforms_box.py:97
Method__call__
(self, results)
mmdet_extension/datasets/pipelines/transforms_box.py:123
Method__call__
(self, bboxes, labels, masks=None)
mmdet_extension/datasets/pipelines/transforms_box.py:176
Method__call__
(self, results)
mmdet_extension/datasets/pipelines/transforms_box.py:184
Method__call__
(self, results)
mmdet_extension/datasets/pipelines/semi_augment.py:22
Method__call__
(self, results)
mmdet_extension/datasets/pipelines/semi_augment.py:38
Method__call__
(self, results)
mmdet_extension/datasets/pipelines/semi_augment.py:55
Method__call__
(self, results)
mmdet_extension/datasets/pipelines/semi_augment.py:70
Method__getitem__
(self, idx)
mmdet_extension/datasets/semi_dataset.py:56
Method__init__
(self, dataloader, start=None, interval=1,
mmdet_extension/core/hooks/semi_eval_hooks.py:62
Method__init__
(self, task_num, bar_width=50)
mmdet_extension/apis/test.py:21
Method__init__
(self, ann_file, pipeline, classes=None, d
mmdet_extension/datasets/ossod_coco.py:18
Method__init__
(self, ann_file, pipeline, classes=None, d
mmdet_extension/datasets/coco_nolabel.py:18
Method__init__
(self, ann_file, pipeline, ann_file_u, pip
mmdet_extension/datasets/semi_dataset.py:18
Method__init__
(self, ann_file, pipeline, classes=None, d
mmdet_extension/datasets/txt_style.py:20
Method__init__
(self, p=0.5, scale=(0.02, 0.33), ratio=(0.3, 3.3), value=0, inplace=False)
mmdet_extension/datasets/pipelines/transforms.py:14
Method__init__
(self, img_shape, scale_factor, bbox_clip_border, scale=None, keep_ratio=True)
mmdet_extension/datasets/pipelines/transforms_box.py:26
Method__init__
(self, img_shape, direction)
mmdet_extension/datasets/pipelines/transforms_box.py:93
Method__init__
(self, x, y, w, h)
mmdet_extension/datasets/pipelines/transforms_box.py:173
Method__init__
(self, use_box=False)
mmdet_extension/datasets/pipelines/semi_augment.py:14
Method__init__
(self)
mmdet_extension/datasets/pipelines/semi_augment.py:29
Method__init__
(self, use_re=True, use_box=False)
mmdet_extension/datasets/pipelines/semi_augment.py:44
Method__init__
(self, use_weak=False, use_re=True, use_box=False)
mmdet_extension/datasets/pipelines/semi_augment.py:64
Method__init__
(self, depth, *args, **kwargs )
mmdet_extension/models/backbones/vgg.py:8
Method__init__
(self, backbone, rpn_head, roi_head, train
mmdet_extension/models/detectors/semi_two_stage.py:15
Method__init__
(self, ema_config=None, ema_ckpt=None, classes=None
mmdet_extension/models/detectors/semi_base.py:26
Method__init__
(self, backbone, rpn_head, roi_head, train
mmdet_extension/models/detectors/unbiased_teacher.py:27
Method__init__
(self, backbone, rpn_head, roi_head, train
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:95
Method__init__
(self, use_sigmoid=False, gamma=2.0, alpha=0.25, reduction='mean', class_weight=None, loss_we
mmdet_extension/models/loss/focal_loss.py:16
Method__init__
(self, fc_out_channels=1024, *args, **kwargs)
mmdet_extension/models/roi_head/bbox_heads/convfc_bbox_head_base.py:64
Method__len__
(self)
mmdet_extension/datasets/ossod_coco.py:34
Method__len__
(self)
mmdet_extension/datasets/coco_nolabel.py:34
Method__len__
(self)
mmdet_extension/datasets/semi_dataset.py:53
Method__len__
(self)
mmdet_extension/datasets/txt_style.py:124
Method__len__
(self)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:90
Method__repr__
(self)
mmdet_extension/models/detectors/unbiased_teacher_ossod.py:87
Method__setattr__
(self, name, value)
mmdet_extension/models/detectors/semi_base.py:136
Method_filter_imgs
(self, min_size=32)
mmdet_extension/datasets/txt_style.py:43
Method_set_group_flag
(self)
mmdet_extension/datasets/txt_style.py:55
Methodafter_train_epoch
(self, runner)
mmdet_extension/core/hooks/semi_eval_hooks.py:50
Methodafter_train_iter
(self, runner)
mmdet_extension/core/hooks/semi_eval_hooks.py:55
Methodbefore_train_epoch
(self, runner)
mmdet_extension/core/hooks/set_epoch_info_hook.py:10
Functioncolor_val_matplotlib
Convert various input in BGR order to normalized RGB matplotlib color tuples. Args: color (:obj`Color` | str | tuple | int | ndarray)
mmdet_extension/core/visualization/image.py:23
next →1–100 of 128, ranked by callers