Types & classes — github.com/TencentYoutuResearch/SceneSegmentation-SCRL

CodeHub MCPcopy Create free account

hub / github.com/TencentYoutuResearch/SceneSegmentation-SCRL / types & classes

Types & classes17 in github.com/TencentYoutuResearch/SceneSegmentation-SCRL

⨍ Functions 115◇ Types & classes 17

↓ 5 callersClassAverageMeter

Computes and stores the average and current value

↓ 4 callersClassGaussianBlur

data/movienet_data.py:88

↓ 2 callersClassBiLSTM

SceneSeg/BiLSTM_protocol.py:6

↓ 2 callersClassMovieNet_SceneSeg_Dataset_Embeddings_Val

SceneSeg/movienet_seg_data.py:118

↓ 2 callersClassMovieNet_Shot_Dataset

data/movienet_data.py:37

↓ 2 callersClassTwoWayTransform

data/movienet_data.py:10

↓ 1 callersClassCluster_GPU

A pytorch GPU implementation for K-Means algorithm, which is used for real-time clutering in SCRL.

cluster/Group.py:7

↓ 1 callersClassEncoder

models/backbones/visual/resnet.py:252

↓ 1 callersClassKMeans_Mixed

This version uses GPU for tensor computation and CPU for indexing to improve the speed of the algorithm.

cluster/Group.py:50

↓ 1 callersClassMovieNet_SceneSeg_Dataset_Embeddings_Train

SceneSeg/movienet_seg_data.py:7

↓ 1 callersClassMovieNet_SingleShot_Dataset

extract_embeddings.py:12

↓ 1 callersClassProgressMeter

↓ 1 callersClassResNet

models/backbones/visual/resnet.py:143

↓ 1 callersClassSCRL

Referenced from MoCo[1] and SCRL[2]. [1] https://arxiv.org/abs/1911.05722 [2] https://arxiv.org/abs/2205.05487

models/core/SCRL_MoCo.py:5

↓ 1 callersClassShotMask

Drop the shot from the middle of a scene

SceneSeg/BiLSTM_protocol.py:67

ClassBasicBlock

models/backbones/visual/resnet.py:36

ClassBottleneck

models/backbones/visual/resnet.py:85