MCPcopy Create free account

hub / github.com/UX-Decoder/Semantic-SAM / types & classes

Types & classes112 in github.com/UX-Decoder/Semantic-SAM

↓ 4 callersClassBaseModel
semantic_sam/BaseModel.py:12
↓ 4 callersClassInteractiveEvaluator
Evaluate point interactive IoU metrics.
datasets/evaluation/interactive_evaluation.py:70
↓ 4 callersClassMLP
Very simple multi-layer perceptron (also called FFN)
semantic_sam/body/decoder/utils/utils.py:11
↓ 3 callersClassGenericMask
Attribute: polygons (list[ndarray]): list[ndarray]: polygons for this mask. Each ndarray has format [x, y, x, y, ...]
utils/visualizer.py:59
↓ 3 callersClassInstanceSegEvaluator
Evaluate AR for object proposals, AP for instance detection/segmentation, AP for keypoint detection outputs using COCO's metrics. See htt
datasets/evaluation/instance_evaluation.py:30
↓ 3 callersClassLayerNorm
semantic_sam/language/LangEncoder/transformer.py:21
↓ 3 callersClassMLP
Very simple multi-layer perceptron (also called FFN)
semantic_sam/body/decoder/modules.py:186
↓ 3 callersClassMaskData
A structure for storing masks and their related data in batched format. Implements basic filtering and concatenation.
utils/sam_utils/amg.py:16
↓ 3 callersClassNestedTensor
semantic_sam/utils/misc.py:42
↓ 3 callersClassSemSegEvaluator
Evaluate semantic segmentation metrics.
datasets/evaluation/segmentation_evaluation.py:21
↓ 2 callersClassCfgNode
The same as `fvcore.common.config.CfgNode`, but different in: 1. Use unsafe yaml loading by default. Note that this may lead to arbit
utils/Config.py:3
↓ 2 callersClassMSDeformAttn
semantic_sam/body/encoder/ops/modules/ms_deform_attn.py:34
↓ 2 callersClassMaskFormerInstanceDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer for instance segmentation.
datasets/dataset_mappers/mask_former_instance_dataset_mapper.py:41
↓ 2 callersClassPascalInstanceNewBaselineDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer. This dataset mapper app
datasets/dataset_mappers/pascal_instance_new_baseline_dataset_mapper.py:93
↓ 2 callersClassPositionEmbeddingSine
This is a more standard version of the position embedding, very similar to the one used by the Attention is all you need paper, generalized t
semantic_sam/modules/position_encoding.py:12
↓ 2 callersClassSamBaselineDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer. This dataset mapper app
datasets/dataset_mappers/sam_baseline_dataset_mapper.py:96
↓ 2 callersClassTSVFile
datasets/utils/tsv/tsv_io.py:12
↓ 2 callersClassTransformerEncoder
semantic_sam/body/transformer_blocks.py:78
↓ 2 callersClassTransformerEncoderLayer
semantic_sam/body/transformer_blocks.py:154
↓ 2 callersClassVisualizer
Visualizer that draws data about detection/segmentation on images. It contains methods like `draw_{text,box,circle,line,binary_mask,polygon}
utils/visualizer.py:331
↓ 1 callersClassBasicLayer
A basic focal modulation layer for one stage. Args: dim (int): Number of feature channels depth (int): Depths of this stage.
semantic_sam/backbone/focal_dw.py:206
↓ 1 callersClassBasicLayer
A basic Swin Transformer layer for one stage. Args: dim (int): Number of feature channels depth (int): Depths of this stage.
semantic_sam/backbone/swin_new.py:340
↓ 1 callersClassBasicLayer
A basic focal modulation layer for one stage. Args: dim (int): Number of feature channels depth (int): Depths of this stage.
semantic_sam/backbone/focal.py:197
↓ 1 callersClassBasicLayer
A basic Swin Transformer layer for one stage. Args: dim (int): Number of feature channels depth (int): Depths of this stage.
semantic_sam/backbone/swin.py:351
↓ 1 callersClassCOCOInstanceNewBaselineDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer. This dataset mapper app
datasets/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py:72
↓ 1 callersClassCOCOInteractivePanopticNewBaselineDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer. This dataset mapper app
datasets/dataset_mappers/coco_interactive_panoptic_new_baseline_dataset_mapper.py:75
↓ 1 callersClassCOCOPanopticEvaluator
Evaluate Panoptic Quality metrics on COCO using PanopticAPI. It saves panoptic segmentation prediction in `output_dir` It contains a syn
datasets/evaluation/panoptic_evaluation.py:24
↓ 1 callersClassCOCOPanopticNewBaselineDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer. This dataset mapper app
datasets/dataset_mappers/coco_panoptic_new_baseline_dataset_mapper.py:54
↓ 1 callersClassCoCoInferenceDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by the model. This is the default call
datasets/dataset_mappers/inference_mapper_with_gt.py:21
↓ 1 callersClassD2FocalNet
semantic_sam/backbone/focal_dw.py:695
↓ 1 callersClassD2FocalNet
semantic_sam/backbone/focal.py:598
↓ 1 callersClassD2SwinTransformer
semantic_sam/backbone/swin.py:769
↓ 1 callersClassDeformableTransformerDecoderLayer
semantic_sam/body/decoder/utils/dino_decoder.py:192
↓ 1 callersClassFileProgressingbar
datasets/utils/tsv/io_common.py:18
↓ 1 callersClassFocalModulation
Focal Modulation Args: dim (int): Number of input channels. proj_drop (float, optional): Dropout ratio of output. Default: 0.0
semantic_sam/backbone/focal_dw.py:44
↓ 1 callersClassFocalModulation
Focal Modulation Args: dim (int): Number of input channels. proj_drop (float, optional): Dropout ratio of output. Default: 0.0
semantic_sam/backbone/focal.py:44
↓ 1 callersClassFocalModulationBlock
Focal Modulation Block. Args: dim (int): Number of input channels. mlp_ratio (float): Ratio of mlp hidden dim to embedding dim.
semantic_sam/backbone/focal_dw.py:118
↓ 1 callersClassFocalModulationBlock
Focal Modulation Block. Args: dim (int): Number of input channels. mlp_ratio (float): Ratio of mlp hidden dim to embedding dim.
semantic_sam/backbone/focal.py:118
↓ 1 callersClassGeneralizedMaskDINO
Main class for mask classification semantic segmentation architectures.
semantic_sam/architectures/interactive_mask_dino.py:27
↓ 1 callersClassIMaskDINODecoder
semantic_sam/body/decoder/interactive_mask_dino.py:24
↓ 1 callersClassIMaskDINOHead
semantic_sam/body/general_head.py:19
↓ 1 callersClassImageMask
demo_auto_generation.py:69
↓ 1 callersClassImageMask
Sets: source="canvas", tool="sketch"
demo.py:66
↓ 1 callersClassImageNetDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer. This dataset mapper app
datasets/dataset_mappers/imagenet_dataset_mapper.py:24
↓ 1 callersClassJointBoxPointInteractiveEvaluator
datasets/evaluation/interactive_evaluation.py:14
↓ 1 callersClassJointLoader
datasets/build.py:62
↓ 1 callersClassLanguageEncoder
semantic_sam/language/vlpencoder.py:19
↓ 1 callersClassLanguageEncoder
semantic_sam/language/encoder.py:13
↓ 1 callersClassM2MHungarianMatcher
This class computes an assignment between the targets and the predictions of the network For efficiency reasons, the targets don't include the no
semantic_sam/modules/many2many_matcher.py:75
↓ 1 callersClassMSDeformAttnTransformerEncoder
semantic_sam/body/encoder/encoder_deform.py:145
↓ 1 callersClassMSDeformAttnTransformerEncoderLayer
semantic_sam/body/encoder/encoder_deform.py:103
↓ 1 callersClassMSDeformAttnTransformerEncoderOnly
semantic_sam/body/encoder/encoder_deform.py:29
↓ 1 callersClassMaskDINOEncoder
This is the multi-scale encoder in detection models, also named as pixel decoder in segmentation models.
semantic_sam/body/encoder/encoder_deform.py:178
↓ 1 callersClassMaskFormerPanopticDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer for panoptic segmentation.
datasets/dataset_mappers/mask_former_panoptic_dataset_mapper.py:21
↓ 1 callersClassMaskFormerPanopticDatasetMapperInteractive
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer for panoptic segmentation.
datasets/dataset_mappers/mask_former_interactive_panoptic_dataset_mapper.py:25
↓ 1 callersClassMaskFormerSemanticDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer for semantic segmentation.
datasets/dataset_mappers/mask_former_semantic_dataset_mapper.py:19
↓ 1 callersClassMlp
Multilayer perceptron.
semantic_sam/backbone/focal_dw.py:24
↓ 1 callersClassMlp
Multilayer perceptron.
semantic_sam/backbone/swin_new.py:21
↓ 1 callersClassMlp
Multilayer perceptron.
semantic_sam/backbone/focal.py:24
↓ 1 callersClassMlp
Multilayer perceptron.
semantic_sam/backbone/swin.py:26
↓ 1 callersClassO365InstanceNewBaselineDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer. This dataset mapper app
datasets/dataset_mappers/o365_instance_new_baseline_dataset_mapper.py:76
↓ 1 callersClassPASCALPARTEvaluator
PASCALPARTEvaluator on open_vocabulary
datasets/evaluation/pascal_part_evaluation.py:32
↓ 1 callersClassPartFilterWholeInstanceNewBaselineDatasetMapper
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer. This dataset mapper app
datasets/dataset_mappers/part_data_filter_whole_new_instance_dataset_mapper.py:111
↓ 1 callersClassPatchEmbed
Image to Patch Embedding Args: patch_size (int): Patch token size. Default: 4. in_chans (int): Number of input image channels. D
semantic_sam/backbone/focal_dw.py:368
↓ 1 callersClassPatchEmbed
Image to Patch Embedding Args: patch_size (int): Patch token size. Default: 4. in_chans (int): Number of input image channels. Def
semantic_sam/backbone/swin_new.py:456
↓ 1 callersClassPatchEmbed
Image to Patch Embedding Args: patch_size (int): Patch token size. Default: 4. in_chans (int): Number of input image channels. D
semantic_sam/backbone/focal.py:287
↓ 1 callersClassPatchEmbed
Image to Patch Embedding Args: patch_size (int): Patch token size. Default: 4. in_chans (int): Number of input image channels. Def
semantic_sam/backbone/swin.py:467
↓ 1 callersClassQuickGELU
semantic_sam/language/LangEncoder/transformer.py:39
↓ 1 callersClassResidualAttentionBlock
semantic_sam/language/LangEncoder/transformer.py:44
↓ 1 callersClassSamBaselineDatasetMapperJSON
A callable which takes a dataset dict in Detectron2 Dataset format, and map it into a format used by MaskFormer. This dataset mapper app
datasets/dataset_mappers/sam_baseline_dataset_mapper_json.py:79
↓ 1 callersClassSemanticSamAutomaticMaskGenerator
tasks/automatic_mask_generator.py:53
↓ 1 callersClassSetCriterion
This class computes the loss for DETR. The process happens in two steps: 1) we compute one to one assignment (max iou score) between groun
semantic_sam/modules/criterion_interactive_many_to_one.py:124
↓ 1 callersClassSetCriterionOsPartWholeM2M
This class computes the loss for DETR. The process happens in two steps: 1) we compute hungarian assignment between ground truth boxes and
semantic_sam/modules/criterion_interactive_many_to_many.py:129
↓ 1 callersClassSwinTransformerBlock
Swin Transformer Block. Args: dim (int): Number of input channels. num_heads (int): Number of attention heads. window_size
semantic_sam/backbone/swin_new.py:174
↓ 1 callersClassSwinTransformerBlock
Swin Transformer Block. Args: dim (int): Number of input channels. num_heads (int): Number of attention heads. window_size
semantic_sam/backbone/swin.py:180
↓ 1 callersClassTrainer
Extension of the Trainer class adapted to MaskFormer.
train_net.py:58
↓ 1 callersClassTransformer
semantic_sam/language/LangEncoder/transformer.py:81
↓ 1 callersClassTransformerDecoder
semantic_sam/body/transformer_blocks.py:105
↓ 1 callersClassTransformerDecoder
semantic_sam/body/decoder/utils/dino_decoder.py:18
↓ 1 callersClassTransformerDecoderLayer
semantic_sam/body/transformer_blocks.py:231
↓ 1 callersClassTransformerEncoderOnly
semantic_sam/body/encoder/transformer_encoder_fpn.py:151
↓ 1 callersClassTransformerEncoderPixelDecoder
semantic_sam/body/encoder/transformer_encoder_fpn.py:193
↓ 1 callersClassVisImage
utils/visualizer.py:257
↓ 1 callersClassWindowAttention
Window based multi-head self attention (W-MSA) module with relative position bias. It supports both of shifted and non-shifted window. Args:
semantic_sam/backbone/swin_new.py:74
↓ 1 callersClassWindowAttention
Window based multi-head self attention (W-MSA) module with relative position bias. It supports both of shifted and non-shifted window. Args:
semantic_sam/backbone/swin.py:79
↓ 1 callersClass_LinearWithBias
semantic_sam/modules/attention.py:324
↓ 1 callersClass_PanopticPrediction
Unify different panoptic annotation/prediction formats
utils/visualizer.py:155
ClassAverageMeter
Computes and stores the average and current value.
utils/misc.py:44
ClassBackbone
Abstract base class for network backbones.
semantic_sam/backbone/backbone.py:11
ClassBasePixelDecoder
semantic_sam/body/encoder/transformer_encoder_fpn.py:22
ClassColorMode
Enum of different color modes to use for instance visualizations.
utils/visualizer.py:37
ClassCrossAttentionLayer
semantic_sam/body/decoder/modules.py:70
ClassD2SwinTransformer
semantic_sam/backbone/swin_new.py:687
ClassDatasetMapperFilterByBox
datasets/dataset_mappers/dataset_mapper_filterbybox.py:35
ClassFFNLayer
semantic_sam/body/decoder/modules.py:132
ClassFocalNet
FocalNet backbone. Args: pretrain_img_size (int): Input image size for training the pretrained model, used in absolute posti
semantic_sam/backbone/focal_dw.py:434
ClassFocalNet
FocalNet backbone. Args: pretrain_img_size (int): Input image size for training the pretrained model, used in absolute posti
semantic_sam/backbone/focal.py:340
ClassFullModelGradientClippingOptimizer
train_net.py:247
ClassFullModelGradientClippingOptimizer
datasets/build.py:655
ClassHungarianMatcher
This class computes an assignment between the targets and the predictions of the network For efficiency reasons, the targets don't include the no
semantic_sam/modules/matcher.py:77
next →1–100 of 112, ranked by callers