├── .gitignore ├── README.md ├── __init__.py ├── ckpts └── ckpt │ ├── log.txt │ └── tagging10_fold │ └── log.txt ├── inference.py ├── inference.sh ├── pre ├── extract_features.py ├── extract_features.sh ├── extract_frame.sh └── extract_video_frame.py ├── pretrain.py ├── pretrain.sh ├── pretrained_models └── bert_base_chinese │ └── 42d4a64dda3243ffeca7ec268d5544122e67d9d06b971608796b483925716512.02ac7d664cff08d793eb00d6aac1d04368a1322435e5fe0a27c70b0b3a85327f.json ├── requirements.txt ├── src ├── dataloaders │ ├── README.md │ ├── __init__.py │ ├── pretrain_dataloader.py │ └── tagging_dataloader.py ├── models │ ├── __init__.py │ ├── audio-base │ │ └── audio_config.json │ ├── bert-base │ │ └── bert_config.json │ ├── cross-base │ │ └── cross_config.json │ ├── decoder-base │ │ └── decoder_config.json │ ├── file_utils.py │ ├── loss.py │ ├── modeling.py │ ├── module_audio.py │ ├── module_bert.py │ ├── module_cross.py │ ├── module_decoder.py │ ├── module_visual.py │ ├── optimization.py │ ├── tokenization.py │ ├── until_config.py │ ├── until_module.py │ └── visual-base │ │ └── visual_config.json └── utils │ ├── .ipynb_checkpoints │ └── tagging_eval-checkpoint.py │ ├── __pycache__ │ ├── tagging_eval.cpython-36.pyc │ ├── tagging_eval.cpython-37.pyc │ └── tagging_eval.cpython-38.pyc │ ├── average_precision_calculator.py │ ├── mean_average_precision_calculator.py │ ├── pr_calculator.py │ ├── pr_calculator_per_tag.py │ ├── tagging_eval.py │ └── train_utlis.py ├── train.py ├── train.sh └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ckpts/ckpt/log.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/ckpts/ckpt/log.txt -------------------------------------------------------------------------------- /ckpts/ckpt/tagging10_fold/log.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/ckpts/ckpt/tagging10_fold/log.txt -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/inference.py -------------------------------------------------------------------------------- /inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/inference.sh -------------------------------------------------------------------------------- /pre/extract_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/pre/extract_features.py -------------------------------------------------------------------------------- /pre/extract_features.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/pre/extract_features.sh -------------------------------------------------------------------------------- /pre/extract_frame.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/pre/extract_frame.sh -------------------------------------------------------------------------------- /pre/extract_video_frame.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/pre/extract_video_frame.py -------------------------------------------------------------------------------- /pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/pretrain.py -------------------------------------------------------------------------------- /pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/pretrain.sh -------------------------------------------------------------------------------- /pretrained_models/bert_base_chinese/42d4a64dda3243ffeca7ec268d5544122e67d9d06b971608796b483925716512.02ac7d664cff08d793eb00d6aac1d04368a1322435e5fe0a27c70b0b3a85327f.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/pretrained_models/bert_base_chinese/42d4a64dda3243ffeca7ec268d5544122e67d9d06b971608796b483925716512.02ac7d664cff08d793eb00d6aac1d04368a1322435e5fe0a27c70b0b3a85327f.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch==1.7.0 2 | tqdm 3 | requests 4 | -------------------------------------------------------------------------------- /src/dataloaders/README.md: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/dataloaders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/dataloaders/pretrain_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/dataloaders/pretrain_dataloader.py -------------------------------------------------------------------------------- /src/dataloaders/tagging_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/dataloaders/tagging_dataloader.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/audio-base/audio_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/audio-base/audio_config.json -------------------------------------------------------------------------------- /src/models/bert-base/bert_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/bert-base/bert_config.json -------------------------------------------------------------------------------- /src/models/cross-base/cross_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/cross-base/cross_config.json -------------------------------------------------------------------------------- /src/models/decoder-base/decoder_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/decoder-base/decoder_config.json -------------------------------------------------------------------------------- /src/models/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/file_utils.py -------------------------------------------------------------------------------- /src/models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/loss.py -------------------------------------------------------------------------------- /src/models/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/modeling.py -------------------------------------------------------------------------------- /src/models/module_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/module_audio.py -------------------------------------------------------------------------------- /src/models/module_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/module_bert.py -------------------------------------------------------------------------------- /src/models/module_cross.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/module_cross.py -------------------------------------------------------------------------------- /src/models/module_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/module_decoder.py -------------------------------------------------------------------------------- /src/models/module_visual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/module_visual.py -------------------------------------------------------------------------------- /src/models/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/optimization.py -------------------------------------------------------------------------------- /src/models/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/tokenization.py -------------------------------------------------------------------------------- /src/models/until_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/until_config.py -------------------------------------------------------------------------------- /src/models/until_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/until_module.py -------------------------------------------------------------------------------- /src/models/visual-base/visual_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/models/visual-base/visual_config.json -------------------------------------------------------------------------------- /src/utils/.ipynb_checkpoints/tagging_eval-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/utils/.ipynb_checkpoints/tagging_eval-checkpoint.py -------------------------------------------------------------------------------- /src/utils/__pycache__/tagging_eval.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/utils/__pycache__/tagging_eval.cpython-36.pyc -------------------------------------------------------------------------------- /src/utils/__pycache__/tagging_eval.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/utils/__pycache__/tagging_eval.cpython-37.pyc -------------------------------------------------------------------------------- /src/utils/__pycache__/tagging_eval.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/utils/__pycache__/tagging_eval.cpython-38.pyc -------------------------------------------------------------------------------- /src/utils/average_precision_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/utils/average_precision_calculator.py -------------------------------------------------------------------------------- /src/utils/mean_average_precision_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/utils/mean_average_precision_calculator.py -------------------------------------------------------------------------------- /src/utils/pr_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/utils/pr_calculator.py -------------------------------------------------------------------------------- /src/utils/pr_calculator_per_tag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/utils/pr_calculator_per_tag.py -------------------------------------------------------------------------------- /src/utils/tagging_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/utils/tagging_eval.py -------------------------------------------------------------------------------- /src/utils/train_utlis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/src/utils/train_utlis.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/train.sh -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kniter1/taac-2021-Rank8/HEAD/util.py --------------------------------------------------------------------------------