Functions — github.com/Okery/PyTorch-Simple-MaskRCNN

CodeHub 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

(self, feature, proposal, image_shape, target)

pytorch_mask_rcnn/model/roi_heads.py:114

pytorch_mask_rcnn/model/rpn.py:20

(self, feature, image_shape, target=None)

pytorch_mask_rcnn/model/rpn.py:78

Methodget_image

pytorch_mask_rcnn/datasets/coco_dataset.py:30

Methodget_target

pytorch_mask_rcnn/datasets/coco_dataset.py:41

Functionmaskrcnn_resnet50

Constructs a Mask R-CNN model with a ResNet-50 backbone. Arguments: pretrained (bool): If True, returns a model pre-trained on C

pytorch_mask_rcnn/model/mask_rcnn.py:233

Methodpin_memory

pytorch_mask_rcnn/datasets/utils.py:29

Functionsave_ckpt

(model, optimizer, epochs, ckpt_path, **kwargs)

pytorch_mask_rcnn/utils.py:9

Show the image, with or without the target. args: images (tensor[B, 3, H, W] or List[tensor[3, H, W]]): RGB channels, value rang

pytorch_mask_rcnn/visualizer.py:20

Functionslow_nms

(box, nms_thresh)

pytorch_mask_rcnn/model/box_ops.py:127

Functiontrain_one_epoch

(model, optimizer, data_loader, device, epoch, args)

pytorch_mask_rcnn/engine.py:13

← previous101–111 of 111, ranked by callers