Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/OpenGVLab/unmasked_teacher
/ types & classes
Types & classes
228 in github.com/OpenGVLab/unmasked_teacher
⨍
Functions
1,198
◇
Types & classes
228
↳
Endpoints
1
↓ 23 callers
Class
BoxList
This class represents a set of bounding boxes. The bounding boxes are represented as a Nx4 Tensor. In order to uniquely determine the bou
single_modality/action_detection/alphaction/structures/bounding_box.py:9
↓ 13 callers
Class
SmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
multi_modality/utils/basic_utils.py:21
↓ 12 callers
Class
Compose
Composes several transforms Args: transforms (list of ``Transform`` objects): list of transforms to compose
single_modality/datasets/video_transforms.py:901
↓ 12 callers
Class
ResNLBlock
single_modality/action_detection/alphaction/modeling/common_blocks.py:104
↓ 8 callers
Class
BertModel
The model can behave as an encoder (with only self-attention) as well as a decoder, in which case a layer of cross-attention is added between
multi_modality/models/backbones/bert/xbert.py:1013
↓ 8 callers
Class
ClipToTensor
Convert a list of m (H x W x C) numpy.ndarrays in the range [0, 255] to a torch.FloatTensor of shape (C x m x H x W) in the range [0, 1.0]
single_modality/datasets/volume_transforms.py:16
↓ 8 callers
Class
MetricLogger
multi_modality/utils/basic_utils.py:84
↓ 8 callers
Class
Normalize
Normalize a clip with mean and standard deviation. Given mean: ``(M1,...,Mn)`` and std: ``(S1,..,Sn)`` for ``n`` channels, this transform will
single_modality/datasets/video_transforms.py:1254
↓ 8 callers
Class
Resize
Resizes a list of (H x W x C) numpy.ndarray to the final size The larger the original image is, the more times it takes to interpolate Arg
single_modality/datasets/video_transforms.py:973
↓ 5 callers
Class
EasyDict
Get attributes >>> d = EasyDict({'foo':3}) >>> d['foo'] 3 >>> d.foo 3 >>> d.bar Traceback (most recent call last):
multi_modality/utils/easydict.py:1
↓ 5 callers
Class
FrozenBatchNorm3d
single_modality/action_detection/alphaction/layers/batch_norm.py:81
↓ 5 callers
Class
NpBoxDict
single_modality/action_detection/alphaction/dataset/datasets/ava.py:30
↓ 5 callers
Class
RandomErasing
Randomly selects a rectangle region in an image and erases its pixels. 'Random Erasing Data Augmentation' by Zhong et al. See https://
single_modality/datasets/random_erasing.py:27
↓ 4 callers
Class
Block
single_modality/models/modeling_finetune.py:101
↓ 4 callers
Class
CenterCrop
Extract center crop at the same location for a list of images Args: size (sequence or int): Desired output size for the crop in format (h,
single_modality/datasets/video_transforms.py:1127
↓ 4 callers
Class
Conv3dBN
single_modality/action_detection/alphaction/modeling/common_blocks.py:6
↓ 4 callers
Class
LateralBlock
single_modality/action_detection/alphaction/modeling/backbone/slowfast.py:92
↓ 4 callers
Class
LayerNorm
Subclass torch's LayerNorm to handle fp16.
multi_modality/models/backbones/vit/clip.py:20
↓ 4 callers
Class
LayerNorm
Subclass torch's LayerNorm to handle fp16.
single_modality/models/clip.py:18
↓ 4 callers
Class
MemoryPool
single_modality/action_detection/alphaction/structures/memory_pool.py:3
↓ 4 callers
Class
NpInfoDict
single_modality/action_detection/alphaction/dataset/datasets/ava.py:13
↓ 4 callers
Class
Registry
A helper class for managing registering modules, it extends a dictionary and provides a register functions. Eg. creeting a registry:
single_modality/action_detection/alphaction/utils/registry.py:8
↓ 4 callers
Class
SmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
single_modality/utils.py:21
↓ 4 callers
Class
VisionTransformer
Vision Transformer with support for patch or hybrid CNN input stage
single_modality/models/modeling_finetune.py:205
↓ 3 callers
Class
MetaLoader
wraps multiple data loader
multi_modality/dataset/dataloader.py:10
↓ 3 callers
Class
PatchEmbed
Image to Patch Embedding
single_modality/models/modeling_finetune.py:132
↓ 3 callers
Class
PretrainVisionTransformer
Vision Transformer with support for patch or hybrid CNN input stage
single_modality/models/modeling_pretrain.py:170
↓ 3 callers
Class
VisionTransformer
multi_modality/models/backbones/vit/clip.py:96
↓ 3 callers
Class
VisionTransformer
Vision Transformer with support for patch or hybrid CNN input stage
single_modality/action_detection/modeling_finetune.py:183
↓ 3 callers
Class
VisionTransformer
single_modality/models/clip.py:99
↓ 2 callers
Class
BertAttention
multi_modality/models/backbones/bert/xbert.py:515
↓ 2 callers
Class
BertLMPredictionHead
multi_modality/models/backbones/bert/xbert.py:846
↓ 2 callers
Class
BertOnlyMLMHead
multi_modality/models/backbones/bert/xbert.py:866
↓ 2 callers
Class
ColorfulFormatter
multi_modality/utils/logger.py:200
↓ 2 callers
Class
GroupRandomHorizontalFlip
single_modality/datasets/transforms.py:48
↓ 2 callers
Class
ObjectDetectionEvaluation
Internal implementation of Pascal object detection metrics.
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/object_detection_evaluation.py:457
↓ 2 callers
Class
PretrainVisionTransformer
Vision Transformer with support for patch or hybrid CNN input stage
single_modality/models/modeling_pretrain_umt.py:169
↓ 2 callers
Class
SmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
single_modality/action_detection/utils.py:21
↓ 2 callers
Class
Stack
single_modality/datasets/transforms.py:189
↓ 2 callers
Class
ToTorchFormatTensor
Converts a PIL.Image (RGB) or numpy.ndarray (H x W x C) in the range [0, 255] to a torch.FloatTensor of shape (C x H x W) in the range [0.0, 1.0]
single_modality/datasets/transforms.py:206
↓ 2 callers
Class
VideoClsDataset
Load your own video classification dataset.
single_modality/datasets/kinetics.py:26
↓ 1 callers
Class
AVAVideoDataset
single_modality/action_detection/data/ava.py:15
↓ 1 callers
Class
ActionAccuracyComputation
single_modality/action_detection/alphaction/modeling/roi_heads/action_head/metric.py:5
↓ 1 callers
Class
ActionDetector
single_modality/action_detection/alphaction/modeling/detector/action_detector.py:7
↓ 1 callers
Class
ActionLossComputation
single_modality/action_detection/alphaction/modeling/roi_heads/action_head/loss.py:6
↓ 1 callers
Class
Attention
multi_modality/models/backbones/vit/vit.py:57
↓ 1 callers
Class
Attention
single_modality/action_detection/modeling_finetune.py:57
↓ 1 callers
Class
Attention
single_modality/models/modeling_finetune.py:55
↓ 1 callers
Class
AugmentOp
Apply for video.
single_modality/datasets/rand_augment.py:337
↓ 1 callers
Class
BasicTokenizer
Constructs a BasicTokenizer that will run basic tokenization (punctuation splitting, lower casing, etc.). Args: do_lower_case (:obj:`
multi_modality/models/backbones/bert/tokenization_bert.py:348
↓ 1 callers
Class
BatchCollator
From a list of samples from the dataset, returns the batched objectimages and targets. This should be passed to the DataLoader
single_modality/action_detection/datasets.py:8
↓ 1 callers
Class
BatchCollator
From a list of samples from the dataset, returns the batched objectimages and targets. This should be passed to the DataLoader
single_modality/action_detection/alphaction/dataset/collate_batch.py:28
↓ 1 callers
Class
BertEmbeddings
Construct the embeddings from word, position and token_type embeddings.
multi_modality/models/backbones/bert/xbert.py:272
↓ 1 callers
Class
BertEncoder
multi_modality/models/backbones/bert/xbert.py:690
↓ 1 callers
Class
BertForPreTrainingOutput
Output type of :class:`~transformers.BertForPreTraining`. Args: loss (`optional`, returned when ``labels`` is provided, ``torch.Float
multi_modality/models/backbones/bert/xbert.py:923
↓ 1 callers
Class
BertIntermediate
multi_modality/models/backbones/bert/xbert.py:570
↓ 1 callers
Class
BertLayer
multi_modality/models/backbones/bert/xbert.py:599
↓ 1 callers
Class
BertOnlyNSPHead
multi_modality/models/backbones/bert/xbert.py:876
↓ 1 callers
Class
BertOutput
multi_modality/models/backbones/bert/xbert.py:585
↓ 1 callers
Class
BertPooler
multi_modality/models/backbones/bert/xbert.py:814
↓ 1 callers
Class
BertPreTrainingHeads
multi_modality/models/backbones/bert/xbert.py:886
↓ 1 callers
Class
BertPredictionHeadTransform
multi_modality/models/backbones/bert/xbert.py:829
↓ 1 callers
Class
BertSelfAttention
multi_modality/models/backbones/bert/xbert.py:337
↓ 1 callers
Class
BertSelfOutput
multi_modality/models/backbones/bert/xbert.py:501
↓ 1 callers
Class
Block
multi_modality/models/backbones/vit/vit.py:103
↓ 1 callers
Class
Block
single_modality/action_detection/modeling_finetune.py:103
↓ 1 callers
Class
Bottleneck
single_modality/action_detection/alphaction/modeling/common_blocks.py:36
↓ 1 callers
Class
Combined3dROIHeads
single_modality/action_detection/alphaction/modeling/roi_heads/roi_heads_3d.py:6
↓ 1 callers
Class
ConcatDataset
Same as torch.utils.dataset.dataset.ConcatDataset, but exposes an extra method for querying the sizes of the image
single_modality/action_detection/alphaction/dataset/datasets/concat_dataset.py:6
↓ 1 callers
Class
DataAugmentationForVideoMAE
single_modality/datasets/build.py:11
↓ 1 callers
Class
DropPath
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).
multi_modality/models/backbones/vit/vit.py:24
↓ 1 callers
Class
DropPath
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).
single_modality/action_detection/modeling_finetune.py:24
↓ 1 callers
Class
DropPath
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).
single_modality/models/modeling_finetune.py:21
↓ 1 callers
Class
FastPath
single_modality/action_detection/alphaction/modeling/backbone/slowfast.py:105
↓ 1 callers
Class
GroupColorJitter
single_modality/datasets/transforms.py:88
↓ 1 callers
Class
GroupMultiScaleCrop
single_modality/datasets/transforms.py:116
↓ 1 callers
Class
GroupNormalize
single_modality/datasets/transforms.py:62
↓ 1 callers
Class
HalfPeriodCosStepLR
single_modality/action_detection/alphaction/solver/lr_scheduler.py:52
↓ 1 callers
Class
InteractionBlock
single_modality/action_detection/alphaction/modeling/roi_heads/action_head/IA_structure.py:11
↓ 1 callers
Class
LayerDecayValueAssigner
single_modality/optim_factory.py:45
↓ 1 callers
Class
LayerDecayValueAssigner
single_modality/action_detection/optim_factory.py:40
↓ 1 callers
Class
Linear_Decoder
multi_modality/models/backbones/vit/vit.py:279
↓ 1 callers
Class
Linear_Decoder
single_modality/models/modeling_pretrain_umt.py:135
↓ 1 callers
Class
MLMLoss
masked language modeling loss.
multi_modality/models/criterions.py:205
↓ 1 callers
Class
MaskedLMOutputWithDistill
multi_modality/models/backbones/bert/xbert.py:1573
↓ 1 callers
Class
MetricLogger
single_modality/action_detection/alphaction/utils/metric_logger.py:40
↓ 1 callers
Class
Mixup
Mixup/Cutmix that applies different params to each element or whole batch Args: mixup_alpha (float): mixup alpha value, mixup is active
single_modality/datasets/mixup.py:90
↓ 1 callers
Class
Mlp
multi_modality/models/backbones/vit/vit.py:38
↓ 1 callers
Class
Mlp
single_modality/action_detection/modeling_finetune.py:39
↓ 1 callers
Class
Mlp
single_modality/models/modeling_finetune.py:35
↓ 1 callers
Class
NLBlock
single_modality/action_detection/alphaction/modeling/nonlocal_block.py:9
↓ 1 callers
Class
PatchEmbed
Image to Patch Embedding
multi_modality/models/backbones/vit/vit.py:134
↓ 1 callers
Class
PatchEmbed
Image to Patch Embedding
single_modality/action_detection/modeling_finetune.py:134
↓ 1 callers
Class
Pooler3d
single_modality/action_detection/alphaction/modeling/poolers.py:7
↓ 1 callers
Class
PostProcessor
single_modality/action_detection/engine_for_finetuning.py:139
↓ 1 callers
Class
PostProcessor
single_modality/action_detection/alphaction/modeling/roi_heads/action_head/inference.py:8
↓ 1 callers
Class
PretrainVisionTransformer
Vision Transformer with support for patch or hybrid CNN input stage
multi_modality/models/backbones/vit/vit.py:291
↓ 1 callers
Class
PretrainVisionTransformerDecoder
Vision Transformer with support for patch or hybrid CNN input stage
single_modality/models/modeling_pretrain.py:105
↓ 1 callers
Class
PretrainVisionTransformerEncoder
Vision Transformer with support for patch or hybrid CNN input stage
multi_modality/models/backbones/vit/vit.py:197
↓ 1 callers
Class
PretrainVisionTransformerEncoder
Vision Transformer with support for patch or hybrid CNN input stage
single_modality/models/modeling_pretrain.py:17
next →
1–100 of 228, ranked by callers