Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Okery/PyTorch-Simple-MaskRCNN
/ types & classes
Types & classes
24 in github.com/Okery/PyTorch-Simple-MaskRCNN
⨍
Functions
111
◇
Types & classes
24
↓ 5 callers
Class
Meter
pytorch_mask_rcnn/utils.py:45
↓ 2 callers
Class
BalancedPositiveNegativeSampler
pytorch_mask_rcnn/model/utils.py:36
↓ 2 callers
Class
BoxCoder
pytorch_mask_rcnn/model/box_ops.py:6
↓ 2 callers
Class
Matcher
pytorch_mask_rcnn/model/utils.py:4
↓ 2 callers
Class
RoIAlign
Performs Region of Interest (RoI) Align operator described in Mask R-CNN
pytorch_mask_rcnn/model/pooler.py:7
↓ 2 callers
Class
VOCDataset
pytorch_mask_rcnn/datasets/voc_dataset.py:57
↓ 1 callers
Class
AnchorGenerator
pytorch_mask_rcnn/model/utils.py:68
↓ 1 callers
Class
COCODataset
pytorch_mask_rcnn/datasets/coco_dataset.py:8
↓ 1 callers
Class
CocoEvaluator
pytorch_mask_rcnn/datasets/coco_eval.py:10
↓ 1 callers
Class
CustomBatch
pytorch_mask_rcnn/datasets/utils.py:22
↓ 1 callers
Class
FastRCNNPredictor
pytorch_mask_rcnn/model/mask_rcnn.py:158
↓ 1 callers
Class
GenericMask
pytorch_mask_rcnn/visualizer.py:73
↓ 1 callers
Class
MaskRCNN
Implements Mask R-CNN. The input image to the model is expected to be a tensor, shape [C, H, W], and should be in 0-1 range. The behavi
pytorch_mask_rcnn/model/mask_rcnn.py:16
↓ 1 callers
Class
MaskRCNNPredictor
pytorch_mask_rcnn/model/mask_rcnn.py:176
↓ 1 callers
Class
RPNHead
pytorch_mask_rcnn/model/rpn.py:9
↓ 1 callers
Class
RegionProposalNetwork
pytorch_mask_rcnn/model/rpn.py:27
↓ 1 callers
Class
ResBackbone
pytorch_mask_rcnn/model/mask_rcnn.py:203
↓ 1 callers
Class
RoIHeads
pytorch_mask_rcnn/model/roi_heads.py:36
↓ 1 callers
Class
TextArea
pytorch_mask_rcnn/utils.py:23
↓ 1 callers
Class
Transformer
pytorch_mask_rcnn/model/transform.py:7
↓ 1 callers
Class
VisImage
pytorch_mask_rcnn/visualizer.py:121
↓ 1 callers
Class
Visualizer
pytorch_mask_rcnn/visualizer.py:167
Class
ColorMode
Enum of different color modes to use for instance visualizations.
pytorch_mask_rcnn/visualizer.py:56
Class
GeneralizedDataset
Main class for Generalized Dataset.
pytorch_mask_rcnn/datasets/generalized_dataset.py:9