MCPcopy Create free account

hub / github.com/MiliLab/UniGeoSeg / types & classes

Types & classes142 in github.com/MiliLab/UniGeoSeg

↓ 14 callersClassConversation
A class that keeps all conversation history.
unigeoseg/conversation.py:16
↓ 7 callersClassMLP
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_utils.py:112
↓ 6 callersClassMLP
Very simple multi-layer perceptron (also called FFN)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:187
↓ 5 callersClassDropPath
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_utils.py:92
↓ 5 callersClassLayerNorm2d
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_utils.py:141
↓ 4 callersClassAttention
An attention layer that allows for downscaling the size of the embedding after projection to queries, keys, and values.
unigeoseg/model/multimodal_encoder/sam2/modeling/sam/transformer.py:190
↓ 4 callersClassAverageMeter
Computes and stores the average and current value
unigeoseg/eval_and_test/eval.py:26
↓ 4 callersClassNestedTensor
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/misc.py:63
↓ 4 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
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/position_encoding.py:12
↓ 4 callersClassSwinTransformer
Swin Transformer backbone. A PyTorch impl of : `Swin Transformer: Hierarchical Vision Transformer using Shifted Windows` - https
unigeoseg/model/multimodal_encoder/swin_trans.py:446
↓ 3 callersClassConfigDict
unigeoseg/model/mask_decoder/Mask2Former_Simplify/configs/config.py:11
↓ 3 callersClassConfigDict
unigeoseg/mask_config/config.py:11
↓ 3 callersClassMaskData
A structure for storing masks and their related data in batched format. Implements basic filtering and concatenation.
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:18
↓ 3 callersClassNuImages
Database class for nuImages to help query and retrieve information from the database.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:25
↓ 2 callersClassADE200kDataset
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/dataset.py:117
↓ 2 callersClassAttentionPooling
unigeoseg/model/language_model/llava_phi.py:45
↓ 2 callersClassCausalOutputWithMask
unigeoseg/model/language_model/llava_phi.py:33
↓ 2 callersClassConfig
A facility for config and config files. It supports common file formats as configs: python/json/yaml. The interface is the same as a dict obje
unigeoseg/model/mask_decoder/Mask2Former_Simplify/configs/config.py:48
↓ 2 callersClassConfig
A facility for config and config files. It supports common file formats as configs: python/json/yaml. The interface is the same as a dict obje
unigeoseg/mask_config/config.py:48
↓ 2 callersClassCrossAttentionLayer
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:70
↓ 2 callersClassFFNLayer
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:133
↓ 2 callersClassMSDeformAttnPixelDecoder
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/msdeformattn.py:166
↓ 2 callersClassMaskFormer
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:41
↓ 2 callersClassMaskFormerModel
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/MaskFormerModel.py:80
↓ 2 callersClassNuImagesDataset
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/dataset.py:248
↓ 2 callersClassPatchEmbed
Image to Patch Embedding.
unigeoseg/model/multimodal_encoder/sam2/modeling/backbones/utils.py:63
↓ 2 callersClassREFER
unigeoseg/eval_and_test/refer.py:40
↓ 2 callersClassRRSISDDataset
unigeoseg/eval_and_test/eval_dataset/RS_val_dataset.py:276
↓ 2 callersClassSegmentation
unigeoseg/model/mask_decoder/Mask2Former_Simplify/Segmentation.py:22
↓ 2 callersClassSelfAttentionLayer
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:12
↓ 2 callersClassStreamToLogger
Fake file-like stream object that redirects writes to a logger instance.
unigeoseg/utils.py:60
↓ 1 callersClassADEVisualize
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/misc.py:174
↓ 1 callersClassAsyncVideoFrameLoader
A list of video frames to be load asynchronously without blocking session start.
unigeoseg/model/multimodal_encoder/sam2/utils/misc.py:104
↓ 1 callersClassBasicLayer
A basic Swin Transformer layer for one stage. Args: dim (int): Number of feature channels depth (int): Depths of this stage.
unigeoseg/model/multimodal_encoder/swin_trans.py:299
↓ 1 callersClassBasicLayer
A basic Swin Transformer layer for one stage. Args: dim (int): Number of feature channels depth (int): Depths of this stage.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/swin.py:341
↓ 1 callersClassCompression_Connector
unigeoseg/model/multimodal_projector/builder.py:381
↓ 1 callersClassConfig
unigeoseg/model/multimodal_projector/builder.py:504
↓ 1 callersClassD2SwinTransformer
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/swin.py:687
↓ 1 callersClassD_Projector
unigeoseg/model/language_model/projector.py:61
↓ 1 callersClassDataCollector
Collate examples for supervised fine-tuning.
unigeoseg/eval_and_test/eval_dataset/RS_val_dataset.py:544
↓ 1 callersClassDatasetAnalyzer
接收一个类似train.md的文件 格式:**/ct_file.nii.gz, */seg_file.nii.gz
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/DatasetAnalyzer.py:21
↓ 1 callersClassDynamicPerceiverAttention
在原有PerceiverAttention基础上添加动态权重机制 保持原有接口,只增加动态调整功能
unigeoseg/model/language_model/lightweight_dynamic_attention.py:10
↓ 1 callersClassEmptySummaryWriter
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/summary.py:47
↓ 1 callersClassEnhancedAttentionPooling
unigeoseg/model/language_model/llava_phi.py:59
↓ 1 callersClassGradientClipType
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/solver.py:16
↓ 1 callersClassHungarianMatcher
This class computes an assignment between the targets and the predictions of the network For efficiency reasons, the targets don't include the no_
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/matcher.py:100
↓ 1 callersClassIdentityMap
unigeoseg/model/multimodal_projector/builder.py:404
↓ 1 callersClassMLP
Very simple multi-layer perceptron (also called FFN)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/maskformer_transformer_decoder.py:109
↓ 1 callersClassMSDeformAttn
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/modules/ms_deform_attn.py:34
↓ 1 callersClassMSDeformAttnTransformerEncoder
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/msdeformattn.py:69
↓ 1 callersClassMSDeformAttnTransformerEncoderLayer
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/msdeformattn.py:27
↓ 1 callersClassMSDeformAttnTransformerEncoderOnly
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/msdeformattn.py:98
↓ 1 callersClassMaskDecoder
unigeoseg/model/multimodal_encoder/sam2/modeling/sam/mask_decoder.py:15
↓ 1 callersClassMaskFormerHead
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/MaskFormerModel.py:23
↓ 1 callersClassMlp
Multilayer perceptron.
unigeoseg/model/multimodal_encoder/swin_trans.py:16
↓ 1 callersClassMlp
Multilayer perceptron.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/swin.py:22
↓ 1 callersClassMultiScaleAttention
unigeoseg/model/multimodal_encoder/hieradet.py:47
↓ 1 callersClassMultiScaleAttention
unigeoseg/model/multimodal_encoder/sam2/modeling/backbones/hieradet.py:39
↓ 1 callersClassMultiScaleBlock
unigeoseg/model/multimodal_encoder/hieradet.py:92
↓ 1 callersClassMultiScaleBlock
unigeoseg/model/multimodal_encoder/sam2/modeling/backbones/hieradet.py:84
↓ 1 callersClassMultiScaleCompressionConnector
unigeoseg/model/multimodal_projector/builder.py:430
↓ 1 callersClassMultiScaleMaskedTransformerDecoder
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:205
↓ 1 callersClassMultiScaleMaskedTransformerDecoderForOPTPreTrain
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:394
↓ 1 callersClassPatchEmbed
Image to Patch Embedding Args: patch_size (int): Patch token size. Default: 4. in_chans (int): Number of input image channel
unigeoseg/model/multimodal_encoder/swin_trans.py:403
↓ 1 callersClassPatchEmbed
Image to Patch Embedding Args: patch_size (int): Patch token size. Default: 4. in_chans (int): Number of input image channels. Def
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/swin.py:457
↓ 1 callersClassPerceiverAttention
unigeoseg/model/language_model/projector.py:18
↓ 1 callersClassPositionEmbeddingRandom
Positional encoding using random spatial frequencies.
unigeoseg/model/multimodal_encoder/sam2/modeling/position_encoding.py:133
↓ 1 callersClassPromptEncoder
unigeoseg/model/multimodal_encoder/sam2/modeling/sam/prompt_encoder.py:17
↓ 1 callersClassRSRSDataset
unigeoseg/eval_and_test/eval_dataset/RS_val_dataset.py:451
↓ 1 callersClassReasonSegDataset
unigeoseg/eval_and_test/eval_dataset/RS_val_dataset.py:351
↓ 1 callersClassRefSegRSDataset
unigeoseg/eval_and_test/eval_dataset/RS_val_dataset.py:203
↓ 1 callersClassResNet
unigeoseg/model/multimodal_projector/builder.py:275
↓ 1 callersClassResNet
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/resnet.py:127
↓ 1 callersClassResNetLayerNorm
unigeoseg/model/multimodal_projector/builder.py:220
↓ 1 callersClassResNetSwin
unigeoseg/model/multimodal_projector/builder.py:329
↓ 1 callersClassResNetWOnorm
unigeoseg/model/multimodal_projector/builder.py:166
↓ 1 callersClassSAM2ImagePredictor
unigeoseg/model/multimodal_encoder/sam2/sam2_image_predictor.py:20
↓ 1 callersClassSAM2Transforms
unigeoseg/model/multimodal_encoder/sam2/utils/transforms.py:15
↓ 1 callersClassSaver
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/summary.py:90
↓ 1 callersClassSetCriterion
This class computes the loss for DETR. The process happens in two steps: 1) we compute hungarian assignment between ground truth boxes and
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:117
↓ 1 callersClassSummaryWriter
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/summary.py:26
↓ 1 callersClassSwinTransformerBlock
Swin Transformer Block. Args: dim (int): Number of input channels. num_heads (int): Number of attention heads. wind
unigeoseg/model/multimodal_encoder/swin_trans.py:152
↓ 1 callersClassSwinTransformerBlock
Swin Transformer Block. Args: dim (int): Number of input channels. num_heads (int): Number of attention heads. window_size
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/swin.py:175
↓ 1 callersClassTransformer
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/transformer.py:19
↓ 1 callersClassTransformerDecoder
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/transformer.py:105
↓ 1 callersClassTransformerDecoderLayer
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/transformer.py:230
↓ 1 callersClassTransformerEncoder
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/transformer.py:78
↓ 1 callersClassTransformerEncoderLayer
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/transformer.py:154
↓ 1 callersClassTwoWayAttentionBlock
unigeoseg/model/multimodal_encoder/sam2/modeling/sam/transformer.py:112
↓ 1 callersClassTwoWayTransformer
unigeoseg/model/multimodal_encoder/sam2/modeling/sam/transformer.py:19
↓ 1 callersClassUniGeoSegCriterion
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:123
↓ 1 callersClassUniGeoSegModel
unigeoseg/model/language_model/llava_phi.py:106
↓ 1 callersClassWindowAttention
Window based multi-head self attention (W-MSA) module with relative position bias. It supports both of shifted and non-shifted window. Ar
unigeoseg/model/multimodal_encoder/swin_trans.py:69
↓ 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:
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/swin.py:75
↓ 1 callersClasshungarian_matcher_PSALM
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:339
ClassAdaptiveLatentFusion
自适应latent融合,在现有latent基础上增加动态性
unigeoseg/model/language_model/lightweight_text_vision_fusion.py:75
ClassBaseDataset
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/dataset.py:50
ClassBasicBlock
unigeoseg/model/multimodal_projector/builder.py:64
ClassBasicBlock
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/resnet.py:57
ClassBasicBlockLayerNorm
unigeoseg/model/multimodal_projector/builder.py:13
next →1–100 of 142, ranked by callers