├── LICENSE ├── README.md ├── configs ├── psl_balanced_chunk_10sec.yaml ├── psl_balanced_chunk_2sec.yaml ├── psl_balanced_chunk_5sec.yaml ├── teacher_student_chunk_10sec.yaml ├── teacher_student_chunk_2sec.yaml └── teacher_student_chunk_5sec.yaml ├── figures └── PSL.png ├── requirements.txt ├── scripts ├── 1_download_audioset.sh ├── 2_prepare_data.sh └── utils │ ├── dump_wave_to_hdf5.py │ └── prepare_label_list.py ├── src ├── average_models_weights.py ├── dataset.py ├── forward.py ├── losses.py ├── models.py ├── run.py └── utils.py └── train_psl.sh /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/README.md -------------------------------------------------------------------------------- /configs/psl_balanced_chunk_10sec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/configs/psl_balanced_chunk_10sec.yaml -------------------------------------------------------------------------------- /configs/psl_balanced_chunk_2sec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/configs/psl_balanced_chunk_2sec.yaml -------------------------------------------------------------------------------- /configs/psl_balanced_chunk_5sec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/configs/psl_balanced_chunk_5sec.yaml -------------------------------------------------------------------------------- /configs/teacher_student_chunk_10sec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/configs/teacher_student_chunk_10sec.yaml -------------------------------------------------------------------------------- /configs/teacher_student_chunk_2sec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/configs/teacher_student_chunk_2sec.yaml -------------------------------------------------------------------------------- /configs/teacher_student_chunk_5sec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/configs/teacher_student_chunk_5sec.yaml -------------------------------------------------------------------------------- /figures/PSL.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/figures/PSL.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/1_download_audioset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/scripts/1_download_audioset.sh -------------------------------------------------------------------------------- /scripts/2_prepare_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/scripts/2_prepare_data.sh -------------------------------------------------------------------------------- /scripts/utils/dump_wave_to_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/scripts/utils/dump_wave_to_hdf5.py -------------------------------------------------------------------------------- /scripts/utils/prepare_label_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/scripts/utils/prepare_label_list.py -------------------------------------------------------------------------------- /src/average_models_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/src/average_models_weights.py -------------------------------------------------------------------------------- /src/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/src/dataset.py -------------------------------------------------------------------------------- /src/forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/src/forward.py -------------------------------------------------------------------------------- /src/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/src/losses.py -------------------------------------------------------------------------------- /src/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/src/models.py -------------------------------------------------------------------------------- /src/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/src/run.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RicherMans/PSL/HEAD/src/utils.py -------------------------------------------------------------------------------- /train_psl.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | python3 src/run.py run $@ 4 | --------------------------------------------------------------------------------