Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/UX-Decoder/Semantic-SAM
/ types & classes
Types & classes
112 in github.com/UX-Decoder/Semantic-SAM
⨍
Functions
759
◇
Types & classes
112
↓ 4 callers
Class
BaseModel
semantic_sam/BaseModel.py:12
↓ 4 callers
Class
InteractiveEvaluator
Evaluate point interactive IoU metrics.
datasets/evaluation/interactive_evaluation.py:70
↓ 4 callers
Class
MLP
Very simple multi-layer perceptron (also called FFN)
semantic_sam/body/decoder/utils/utils.py:11
↓ 3 callers
Class
GenericMask
Attribute: polygons (list[ndarray]): list[ndarray]: polygons for this mask. Each ndarray has format [x, y, x, y, ...]
utils/visualizer.py:59
↓ 3 callers
Class
InstanceSegEvaluator
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 callers
Class
LayerNorm
semantic_sam/language/LangEncoder/transformer.py:21
↓ 3 callers
Class
MLP
Very simple multi-layer perceptron (also called FFN)
semantic_sam/body/decoder/modules.py:186
↓ 3 callers
Class
MaskData
A structure for storing masks and their related data in batched format. Implements basic filtering and concatenation.
utils/sam_utils/amg.py:16
↓ 3 callers
Class
NestedTensor
semantic_sam/utils/misc.py:42
↓ 3 callers
Class
SemSegEvaluator
Evaluate semantic segmentation metrics.
datasets/evaluation/segmentation_evaluation.py:21
↓ 2 callers
Class
CfgNode
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 callers
Class
MSDeformAttn
semantic_sam/body/encoder/ops/modules/ms_deform_attn.py:34
↓ 2 callers
Class
MaskFormerInstanceDatasetMapper
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 callers
Class
PascalInstanceNewBaselineDatasetMapper
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 callers
Class
PositionEmbeddingSine
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 callers
Class
SamBaselineDatasetMapper
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 callers
Class
TSVFile
datasets/utils/tsv/tsv_io.py:12
↓ 2 callers
Class
TransformerEncoder
semantic_sam/body/transformer_blocks.py:78
↓ 2 callers
Class
TransformerEncoderLayer
semantic_sam/body/transformer_blocks.py:154
↓ 2 callers
Class
Visualizer
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 callers
Class
BasicLayer
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 callers
Class
BasicLayer
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 callers
Class
BasicLayer
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 callers
Class
BasicLayer
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 callers
Class
COCOInstanceNewBaselineDatasetMapper
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 callers
Class
COCOInteractivePanopticNewBaselineDatasetMapper
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 callers
Class
COCOPanopticEvaluator
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 callers
Class
COCOPanopticNewBaselineDatasetMapper
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 callers
Class
CoCoInferenceDatasetMapper
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 callers
Class
D2FocalNet
semantic_sam/backbone/focal_dw.py:695
↓ 1 callers
Class
D2FocalNet
semantic_sam/backbone/focal.py:598
↓ 1 callers
Class
D2SwinTransformer
semantic_sam/backbone/swin.py:769
↓ 1 callers
Class
DeformableTransformerDecoderLayer
semantic_sam/body/decoder/utils/dino_decoder.py:192
↓ 1 callers
Class
FileProgressingbar
datasets/utils/tsv/io_common.py:18
↓ 1 callers
Class
FocalModulation
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 callers
Class
FocalModulation
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 callers
Class
FocalModulationBlock
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 callers
Class
FocalModulationBlock
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 callers
Class
GeneralizedMaskDINO
Main class for mask classification semantic segmentation architectures.
semantic_sam/architectures/interactive_mask_dino.py:27
↓ 1 callers
Class
IMaskDINODecoder
semantic_sam/body/decoder/interactive_mask_dino.py:24
↓ 1 callers
Class
IMaskDINOHead
semantic_sam/body/general_head.py:19
↓ 1 callers
Class
ImageMask
demo_auto_generation.py:69
↓ 1 callers
Class
ImageMask
Sets: source="canvas", tool="sketch"
demo.py:66
↓ 1 callers
Class
ImageNetDatasetMapper
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 callers
Class
JointBoxPointInteractiveEvaluator
datasets/evaluation/interactive_evaluation.py:14
↓ 1 callers
Class
JointLoader
datasets/build.py:62
↓ 1 callers
Class
LanguageEncoder
semantic_sam/language/vlpencoder.py:19
↓ 1 callers
Class
LanguageEncoder
semantic_sam/language/encoder.py:13
↓ 1 callers
Class
M2MHungarianMatcher
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 callers
Class
MSDeformAttnTransformerEncoder
semantic_sam/body/encoder/encoder_deform.py:145
↓ 1 callers
Class
MSDeformAttnTransformerEncoderLayer
semantic_sam/body/encoder/encoder_deform.py:103
↓ 1 callers
Class
MSDeformAttnTransformerEncoderOnly
semantic_sam/body/encoder/encoder_deform.py:29
↓ 1 callers
Class
MaskDINOEncoder
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 callers
Class
MaskFormerPanopticDatasetMapper
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 callers
Class
MaskFormerPanopticDatasetMapperInteractive
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 callers
Class
MaskFormerSemanticDatasetMapper
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 callers
Class
Mlp
Multilayer perceptron.
semantic_sam/backbone/focal_dw.py:24
↓ 1 callers
Class
Mlp
Multilayer perceptron.
semantic_sam/backbone/swin_new.py:21
↓ 1 callers
Class
Mlp
Multilayer perceptron.
semantic_sam/backbone/focal.py:24
↓ 1 callers
Class
Mlp
Multilayer perceptron.
semantic_sam/backbone/swin.py:26
↓ 1 callers
Class
O365InstanceNewBaselineDatasetMapper
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 callers
Class
PASCALPARTEvaluator
PASCALPARTEvaluator on open_vocabulary
datasets/evaluation/pascal_part_evaluation.py:32
↓ 1 callers
Class
PartFilterWholeInstanceNewBaselineDatasetMapper
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 callers
Class
PatchEmbed
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 callers
Class
PatchEmbed
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 callers
Class
PatchEmbed
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 callers
Class
PatchEmbed
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 callers
Class
QuickGELU
semantic_sam/language/LangEncoder/transformer.py:39
↓ 1 callers
Class
ResidualAttentionBlock
semantic_sam/language/LangEncoder/transformer.py:44
↓ 1 callers
Class
SamBaselineDatasetMapperJSON
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 callers
Class
SemanticSamAutomaticMaskGenerator
tasks/automatic_mask_generator.py:53
↓ 1 callers
Class
SetCriterion
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 callers
Class
SetCriterionOsPartWholeM2M
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 callers
Class
SwinTransformerBlock
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 callers
Class
SwinTransformerBlock
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 callers
Class
Trainer
Extension of the Trainer class adapted to MaskFormer.
train_net.py:58
↓ 1 callers
Class
Transformer
semantic_sam/language/LangEncoder/transformer.py:81
↓ 1 callers
Class
TransformerDecoder
semantic_sam/body/transformer_blocks.py:105
↓ 1 callers
Class
TransformerDecoder
semantic_sam/body/decoder/utils/dino_decoder.py:18
↓ 1 callers
Class
TransformerDecoderLayer
semantic_sam/body/transformer_blocks.py:231
↓ 1 callers
Class
TransformerEncoderOnly
semantic_sam/body/encoder/transformer_encoder_fpn.py:151
↓ 1 callers
Class
TransformerEncoderPixelDecoder
semantic_sam/body/encoder/transformer_encoder_fpn.py:193
↓ 1 callers
Class
VisImage
utils/visualizer.py:257
↓ 1 callers
Class
WindowAttention
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 callers
Class
WindowAttention
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 callers
Class
_LinearWithBias
semantic_sam/modules/attention.py:324
↓ 1 callers
Class
_PanopticPrediction
Unify different panoptic annotation/prediction formats
utils/visualizer.py:155
Class
AverageMeter
Computes and stores the average and current value.
utils/misc.py:44
Class
Backbone
Abstract base class for network backbones.
semantic_sam/backbone/backbone.py:11
Class
BasePixelDecoder
semantic_sam/body/encoder/transformer_encoder_fpn.py:22
Class
ColorMode
Enum of different color modes to use for instance visualizations.
utils/visualizer.py:37
Class
CrossAttentionLayer
semantic_sam/body/decoder/modules.py:70
Class
D2SwinTransformer
semantic_sam/backbone/swin_new.py:687
Class
DatasetMapperFilterByBox
datasets/dataset_mappers/dataset_mapper_filterbybox.py:35
Class
FFNLayer
semantic_sam/body/decoder/modules.py:132
Class
FocalNet
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
Class
FocalNet
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
Class
FullModelGradientClippingOptimizer
train_net.py:247
Class
FullModelGradientClippingOptimizer
datasets/build.py:655
Class
HungarianMatcher
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