├── .gitignore ├── LICENSE ├── README.md ├── SceneSeg ├── BiLSTM_protocol.py ├── __init__.py ├── main.py └── movienet_seg_data.py ├── cluster ├── Group.py └── cluster_test.ipynb ├── config ├── SCRL_pretrain_default.yaml ├── SCRL_pretrain_with_imagenet1k.yaml └── SCRL_pretrain_without_imagenet1k.yaml ├── data ├── MovieNet_1.0_shotinfo.json ├── MovieNet_shot_num.json ├── data_preparation.py ├── movie1K.scene_seg_318_name_index_shotnum_label.v1.json ├── movie1K.split.v1.json └── movienet_data.py ├── extract_embeddings.py ├── figures └── puzzle_example.jpg ├── models ├── __init__.py ├── backbones │ ├── __init__.py │ └── visual │ │ └── resnet.py ├── core │ ├── SCRL_MoCo.py │ └── __init__.py └── factory.py ├── pretrain_main.py ├── pretrain_trainer.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/README.md -------------------------------------------------------------------------------- /SceneSeg/BiLSTM_protocol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/SceneSeg/BiLSTM_protocol.py -------------------------------------------------------------------------------- /SceneSeg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /SceneSeg/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/SceneSeg/main.py -------------------------------------------------------------------------------- /SceneSeg/movienet_seg_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/SceneSeg/movienet_seg_data.py -------------------------------------------------------------------------------- /cluster/Group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/cluster/Group.py -------------------------------------------------------------------------------- /cluster/cluster_test.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/cluster/cluster_test.ipynb -------------------------------------------------------------------------------- /config/SCRL_pretrain_default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/config/SCRL_pretrain_default.yaml -------------------------------------------------------------------------------- /config/SCRL_pretrain_with_imagenet1k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/config/SCRL_pretrain_with_imagenet1k.yaml -------------------------------------------------------------------------------- /config/SCRL_pretrain_without_imagenet1k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/config/SCRL_pretrain_without_imagenet1k.yaml -------------------------------------------------------------------------------- /data/MovieNet_1.0_shotinfo.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/data/MovieNet_1.0_shotinfo.json -------------------------------------------------------------------------------- /data/MovieNet_shot_num.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/data/MovieNet_shot_num.json -------------------------------------------------------------------------------- /data/data_preparation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/data/data_preparation.py -------------------------------------------------------------------------------- /data/movie1K.scene_seg_318_name_index_shotnum_label.v1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/data/movie1K.scene_seg_318_name_index_shotnum_label.v1.json -------------------------------------------------------------------------------- /data/movie1K.split.v1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/data/movie1K.split.v1.json -------------------------------------------------------------------------------- /data/movienet_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/data/movienet_data.py -------------------------------------------------------------------------------- /extract_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/extract_embeddings.py -------------------------------------------------------------------------------- /figures/puzzle_example.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/figures/puzzle_example.jpg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/backbones/visual/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/models/backbones/visual/resnet.py -------------------------------------------------------------------------------- /models/core/SCRL_MoCo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/models/core/SCRL_MoCo.py -------------------------------------------------------------------------------- /models/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/models/factory.py -------------------------------------------------------------------------------- /pretrain_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/pretrain_main.py -------------------------------------------------------------------------------- /pretrain_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/pretrain_trainer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentYoutuResearch/SceneSegmentation-SCRL/HEAD/utils.py --------------------------------------------------------------------------------