MCPcopy Create free account

hub / github.com/Okery/PyTorch-Simple-MaskRCNN / functions

Functions111 in github.com/Okery/PyTorch-Simple-MaskRCNN

⨍ Functions 111◇ Types & classes 24
↓ 9 callersMethodupdate
(self, val, n=1)
pytorch_mask_rcnn/utils.py:56
↓ 4 callersMethoddecode
From a set of original boxes and encoded relative box offsets, get the decoded boxes. Arguments: delta (Tensor[N
pytorch_mask_rcnn/model/box_ops.py:39
↓ 4 callersMethodencode
Encode a set of proposals with respect to some reference boxes Arguments: reference_boxes (Tensor[N, 4]): refere
pytorch_mask_rcnn/model/box_ops.py:11
↓ 3 callersMethod__init__
(self, backbone, num_classes, # RPN parameters rpn_fg_iou_thresh=0.7, rpn_bg
pytorch_mask_rcnn/model/mask_rcnn.py:81
↓ 3 callersFunctionbox_iou
Arguments: boxe_a (Tensor[N, 4]) boxe_b (Tensor[M, 4]) Returns: iou (Tensor[N, M]): the NxM matrix containing the pa
pytorch_mask_rcnn/model/box_ops.py:76
↓ 3 callersMethodsave
(self, filepath)
pytorch_mask_rcnn/visualizer.py:154
↓ 2 callersMethod_change_color_brightness
(self, color, brightness_factor)
pytorch_mask_rcnn/visualizer.py:388
↓ 2 callersMethodcheck_dataset
use multithreads to accelerate the process. check the dataset to avoid some problems listed in method `_check`.
pytorch_mask_rcnn/datasets/generalized_dataset.py:28
↓ 2 callersMethoddraw_instance_predictions
(self, predictions, class_names=None, thing_colors=None)
pytorch_mask_rcnn/visualizer.py:183
↓ 2 callersMethodget_target
(self, img_id)
pytorch_mask_rcnn/datasets/voc_dataset.py:96
↓ 2 callersFunctionnms
Arguments: box (Tensor[N, 4]) score (Tensor[N]): scores of the boxes. threshold (float): iou threshold. Returns:
pytorch_mask_rcnn/model/box_ops.py:112
↓ 2 callersFunctionprocess_box
Clip boxes in the image size and remove boxes which are too small.
pytorch_mask_rcnn/model/box_ops.py:98
↓ 2 callersFunctionroi_align
(features, rois, spatial_scale, pooled_height, pooled_width, sampling_ratio)
pytorch_mask_rcnn/model/utils.py:59
↓ 1 callersMethod__init__
(self, in_channels, num_anchors)
pytorch_mask_rcnn/model/rpn.py:10
↓ 1 callersFunction_create_text_labels
(classes, scores, class_names)
pytorch_mask_rcnn/visualizer.py:106
↓ 1 callersMethod_jitter
(self, color)
pytorch_mask_rcnn/visualizer.py:380
↓ 1 callersMethod_setup_figure
(self, img)
pytorch_mask_rcnn/visualizer.py:133
↓ 1 callersMethodaccumulate
(self, coco_results)
pytorch_mask_rcnn/datasets/coco_eval.py:24
↓ 1 callersMethodarea
(self)
pytorch_mask_rcnn/visualizer.py:94
↓ 1 callersMethodbatched_image
(self, image, stride=32)
pytorch_mask_rcnn/model/transform.py:54
↓ 1 callersMethodbbox
(self)
pytorch_mask_rcnn/visualizer.py:97
↓ 1 callersMethodcached_grid_anchor
(self, grid_size, stride)
pytorch_mask_rcnn/model/utils.py:103
↓ 1 callersMethodcompute_loss
(self, objectness, pred_bbox_delta, gt_box, anchor)
pytorch_mask_rcnn/model/rpn.py:65
↓ 1 callersMethodconvert_to_coco_format
(self, overwrite=False)
pytorch_mask_rcnn/datasets/voc_dataset.py:130
↓ 1 callersMethodconvert_to_xyxy
(boxes)
pytorch_mask_rcnn/datasets/coco_dataset.py:37
↓ 1 callersMethodcreate_proposal
(self, anchor, objectness, pred_bbox_delta, image_shape)
pytorch_mask_rcnn/model/rpn.py:47
↓ 1 callersMethoddraw_box
(self, box_coord, alpha=0.5, edge_color="g", line_style="-")
pytorch_mask_rcnn/visualizer.py:327
↓ 1 callersMethoddraw_polygon
(self, segment, color, edge_color=None, alpha=0.5)
pytorch_mask_rcnn/visualizer.py:348
↓ 1 callersMethoddraw_text
(self, text, position, font_size=None, color="g", horizontal_alignment="center")
pytorch_mask_rcnn/visualizer.py:302
↓ 1 callersFunctionexpand_detection
(mask, box, padding)
pytorch_mask_rcnn/model/transform.py:78
↓ 1 callersMethodfastrcnn_inference
(self, class_logit, box_regression, proposal, image_shape)
pytorch_mask_rcnn/model/roi_heads.py:84
↓ 1 callersFunctionfastrcnn_loss
(class_logit, box_regression, label, regression_target)
pytorch_mask_rcnn/model/roi_heads.py:10
↓ 1 callersFunctiongenerate_results
(model, data_loader, device, args)
pytorch_mask_rcnn/engine.py:88
↓ 1 callersMethodget_AP
(self)
pytorch_mask_rcnn/utils.py:33
↓ 1 callersFunctionget_gpu_prop
(show=False)
pytorch_mask_rcnn/gpu.py:11
↓ 1 callersMethodget_image
(self)
pytorch_mask_rcnn/visualizer.py:157
↓ 1 callersMethodget_image
(self, img_id)
pytorch_mask_rcnn/datasets/voc_dataset.py:92
↓ 1 callersMethodgrid_anchor
(self, grid_size, stride)
pytorch_mask_rcnn/model/utils.py:90
↓ 1 callersMethodhas_mask
(self)
pytorch_mask_rcnn/model/roi_heads.py:58
↓ 1 callersFunctionmain
(args)
train.py:12
↓ 1 callersMethodmake_aspect_ratios
(self)
pytorch_mask_rcnn/datasets/voc_dataset.py:82
↓ 1 callersMethodmask_to_polygons
(self, mask)
pytorch_mask_rcnn/visualizer.py:82
↓ 1 callersFunctionmaskrcnn_loss
(mask_logit, proposal, matched_idx, label, gt_mask)
pytorch_mask_rcnn/model/roi_heads.py:23
↓ 1 callersMethodnormalize
(self, image)
pytorch_mask_rcnn/model/transform.py:21
↓ 1 callersMethodoverlay_instances
(self, boxes=None, labels=None, masks=None, assigned_colors=None, alpha=0.5)
pytorch_mask_rcnn/visualizer.py:209
↓ 1 callersFunctionpaste_masks_in_image
(mask, box, padding, image_shape)
pytorch_mask_rcnn/model/transform.py:99
↓ 1 callersMethodpostprocess
(self, result, image_shape, ori_image_shape)
pytorch_mask_rcnn/model/transform.py:64
↓ 1 callersFunctionprepare_for_coco
(predictions)
pytorch_mask_rcnn/datasets/coco_eval.py:49
↓ 1 callersFunctionrandom_color
(rgb=False, maximum=255)
pytorch_mask_rcnn/visualizer.py:398
↓ 1 callersMethodreset
(self)
pytorch_mask_rcnn/utils.py:50
↓ 1 callersMethodreset_image
(self, img)
pytorch_mask_rcnn/visualizer.py:150
↓ 1 callersMethodresize
(self, image, target)
pytorch_mask_rcnn/model/transform.py:30
↓ 1 callersMethodselect_training_samples
(self, proposal, target)
pytorch_mask_rcnn/model/roi_heads.py:65
↓ 1 callersMethodset_cell_anchor
(self, dtype, device)
pytorch_mask_rcnn/model/utils.py:76
↓ 1 callersMethodsetup_scale
(self, feature_shape, image_shape)
pytorch_mask_rcnn/model/pooler.py:29
↓ 1 callersMethodshow
(self, title=None)
pytorch_mask_rcnn/visualizer.py:367
↓ 1 callersFunctionsort
(d, tmp={})
pytorch_mask_rcnn/gpu.py:32
↓ 1 callersMethodsummarize
(self)
pytorch_mask_rcnn/datasets/coco_eval.py:40
↓ 1 callersFunctiontarget_to_coco_ann
(target)
pytorch_mask_rcnn/datasets/voc_dataset.py:22
↓ 1 callersMethodwrite
(self, s)
pytorch_mask_rcnn/utils.py:27
Method__call__
(self, *args, **kwargs)
pytorch_mask_rcnn/visualizer.py:180
Method__call__
Arguments: feature (Tensor[N, C, H, W]) proposal (Tensor[K, 4]) image_shape (Torch.Size([H, W]))
pytorch_mask_rcnn/model/pooler.py:40
Method__call__
Arguments: iou (Tensor[M, N]): containing the pairwise quality between M ground-truth boxes and N predicted boxes.
pytorch_mask_rcnn/model/utils.py:10
Method__call__
(self, label)
pytorch_mask_rcnn/model/utils.py:41
Method__call__
(self, feature, image_size)
pytorch_mask_rcnn/model/utils.py:114
Method__call__
(self, image, target)
pytorch_mask_rcnn/model/transform.py:14
Method__getitem__
(self, i)
pytorch_mask_rcnn/datasets/generalized_dataset.py:18
Method__init__
(self)
pytorch_mask_rcnn/utils.py:24
Method__init__
(self, name)
pytorch_mask_rcnn/utils.py:46
Method__init__
(self, mask, height, width)
pytorch_mask_rcnn/visualizer.py:74
Method__init__
Args: img (ndarray): an RGB image of shape (H, W, 3) in range [0, 255]. scale (float): scale the input image
pytorch_mask_rcnn/visualizer.py:122
Method__init__
(self, img_rgb, scale=1.0, instance_mode=ColorMode.IMAGE)
pytorch_mask_rcnn/visualizer.py:169
Method__init__
Arguments: output_size (Tuple[int, int]): the size of the output after the cropping is performed, as (height, wid
pytorch_mask_rcnn/model/pooler.py:13
Method__init__
(self, in_channels, mid_channels, num_classes)
pytorch_mask_rcnn/model/mask_rcnn.py:159
Method__init__
Arguments: in_channels (int) layers (Tuple[int]) dim_reduced (int) num_classes (int)
pytorch_mask_rcnn/model/mask_rcnn.py:177
Method__init__
(self, backbone_name, pretrained)
pytorch_mask_rcnn/model/mask_rcnn.py:204
Method__init__
(self, high_threshold, low_threshold, allow_low_quality_matches=False)
pytorch_mask_rcnn/model/utils.py:5
Method__init__
(self, num_samples, positive_fraction)
pytorch_mask_rcnn/model/utils.py:37
Method__init__
(self, sizes, ratios)
pytorch_mask_rcnn/model/utils.py:69
Method__init__
(self, box_roi_pool, box_predictor, fg_iou_thresh, bg_iou_thresh, num_sample
pytorch_mask_rcnn/model/roi_heads.py:37
Method__init__
(self, min_size, max_size, image_mean, image_std)
pytorch_mask_rcnn/model/transform.py:8
Method__init__
(self, weights, bbox_xform_clip=math.log(1000. / 16))
pytorch_mask_rcnn/model/box_ops.py:7
Method__init__
(self, anchor_generator, head, fg_iou_thresh, bg_iou_thresh, num_samples, p
pytorch_mask_rcnn/model/rpn.py:28
Method__init__
(self, data)
pytorch_mask_rcnn/datasets/utils.py:23
Method__init__
(self, coco_gt, iou_types="bbox")
pytorch_mask_rcnn/datasets/coco_eval.py:11
Method__init__
(self, max_workers=2, verbose=False)
pytorch_mask_rcnn/datasets/generalized_dataset.py:14
Method__init__
(self, data_dir, split, train=False)
pytorch_mask_rcnn/datasets/voc_dataset.py:59
Method__init__
(self, data_dir, split, train=False)
pytorch_mask_rcnn/datasets/coco_dataset.py:9
Method__len__
(self)
pytorch_mask_rcnn/datasets/generalized_dataset.py:25
Method__str__
(self)
pytorch_mask_rcnn/utils.py:30
Method__str__
(self)
pytorch_mask_rcnn/utils.py:62
Method_check
(self, seq)
pytorch_mask_rcnn/datasets/generalized_dataset.py:62
Methodcoco
(self)
pytorch_mask_rcnn/datasets/voc_dataset.py:123
Functioncollate_wrapper
(batch)
pytorch_mask_rcnn/datasets/utils.py:18
Functioncollect_gpu_info
(model_name, fps)
pytorch_mask_rcnn/gpu.py:42
Functiondatasets
(ds, *args, **kwargs)
pytorch_mask_rcnn/datasets/utils.py:7
Functionevaluate
(model, data_loader, device, args, generate=True)
pytorch_mask_rcnn/engine.py:59
Methodforward
(self, image, target=None)
pytorch_mask_rcnn/model/mask_rcnn.py:141
Methodforward
(self, x)
pytorch_mask_rcnn/model/mask_rcnn.py:166
Methodforward
(self, x)
pytorch_mask_rcnn/model/mask_rcnn.py:225
next →1–100 of 111, ranked by callers