├── .gitignore ├── README.md ├── augmentation.py ├── cassle_BarlowTwins.py ├── cassle_moco.py ├── cassle_simclr.py ├── cl_table_tools.py ├── confusion_matrix_fig.py ├── dataset ├── cl_pipeline.py ├── data_pipelines.py ├── dataio_test.ipynb ├── label_encoder_sb.txt ├── label_encoder_tauuas2019_ordered.txt ├── label_encoder_us8k_ordered.txt ├── label_encoder_vggsound_ordered.txt ├── prepare_tauuas2019.py ├── prepare_urbansound8k.py └── prepare_vggsound.py ├── debug_augmentation.ipynb ├── hparams ├── tau19 │ ├── cassle_BarlowTwins.yaml │ ├── cassle_moco.yaml │ ├── cassle_simclr.yaml │ ├── knn_eval.yaml │ ├── knn_eval_10task.yaml │ ├── knn_eval_moco-Copy1.yaml │ ├── knn_eval_moco.yaml │ ├── linclf_train.yaml │ ├── linclf_train_10task.yaml │ ├── linclf_train_moco.yaml │ ├── supBarlowTwins_train.yaml │ ├── sup_labeldist.yaml │ ├── sup_train.yaml │ ├── supclr_train.yaml │ ├── supclr_train_10task.yaml │ └── supmoco_train.yaml ├── us8k │ ├── cassiam_train.yaml │ ├── cassle_BarlowTwins.yaml │ ├── cassle_simclr.yaml │ ├── debug.yaml │ ├── knn_eval.yaml │ ├── knn_eval_10task.yaml │ ├── knn_eval_moco.yaml │ ├── linclf_train.yaml │ ├── linclf_train_10task.yaml │ ├── linclf_train_moco.yaml │ ├── moco_train.yaml │ ├── simclr_train.yaml │ ├── simsiam_train.yaml │ ├── supBarlowTwins_train.yaml │ ├── sup_labeldist.yaml │ ├── sup_train.yaml │ ├── supclr_train.yaml │ ├── supclr_train_10task.yaml │ ├── supclr_train_bak.yaml │ ├── supmoco_train.yaml │ ├── supsiam_train.yaml │ ├── train_ecapa_tdnn.yaml │ └── visualize.yaml └── vgg │ ├── linclf_train.yaml │ ├── supBarlowTwins_train.yaml │ ├── sup_train.yaml │ ├── supclr_train.yaml │ └── supmoco_train.yaml ├── knn_eval.py ├── linclf_train.py ├── losses.py ├── moco_train.py ├── models ├── frontend.py ├── modules.py ├── modules_moco.py ├── pann.py ├── pann_moco.py └── tdnn.py ├── schedulers.py ├── simclr_train.py ├── simsiam_debug.py ├── simsiam_train.py ├── supBarlowTwins_train.py ├── sup_labeldist.py ├── sup_train.py ├── supclr_train.py ├── supclr_train_agem.py ├── supclr_train_bak.py ├── supmoco_train.py ├── supsiam_train.py ├── train.py ├── utils.py └── visualize_embedding.ipynb /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/README.md -------------------------------------------------------------------------------- /augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/augmentation.py -------------------------------------------------------------------------------- /cassle_BarlowTwins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/cassle_BarlowTwins.py -------------------------------------------------------------------------------- /cassle_moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/cassle_moco.py -------------------------------------------------------------------------------- /cassle_simclr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/cassle_simclr.py -------------------------------------------------------------------------------- /cl_table_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/cl_table_tools.py -------------------------------------------------------------------------------- /confusion_matrix_fig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/confusion_matrix_fig.py -------------------------------------------------------------------------------- /dataset/cl_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/dataset/cl_pipeline.py -------------------------------------------------------------------------------- /dataset/data_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/dataset/data_pipelines.py -------------------------------------------------------------------------------- /dataset/dataio_test.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/dataset/dataio_test.ipynb -------------------------------------------------------------------------------- /dataset/label_encoder_sb.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/dataset/label_encoder_sb.txt -------------------------------------------------------------------------------- /dataset/label_encoder_tauuas2019_ordered.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/dataset/label_encoder_tauuas2019_ordered.txt -------------------------------------------------------------------------------- /dataset/label_encoder_us8k_ordered.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/dataset/label_encoder_us8k_ordered.txt -------------------------------------------------------------------------------- /dataset/label_encoder_vggsound_ordered.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/dataset/label_encoder_vggsound_ordered.txt -------------------------------------------------------------------------------- /dataset/prepare_tauuas2019.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/dataset/prepare_tauuas2019.py -------------------------------------------------------------------------------- /dataset/prepare_urbansound8k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/dataset/prepare_urbansound8k.py -------------------------------------------------------------------------------- /dataset/prepare_vggsound.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/dataset/prepare_vggsound.py -------------------------------------------------------------------------------- /debug_augmentation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/debug_augmentation.ipynb -------------------------------------------------------------------------------- /hparams/tau19/cassle_BarlowTwins.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/cassle_BarlowTwins.yaml -------------------------------------------------------------------------------- /hparams/tau19/cassle_moco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/cassle_moco.yaml -------------------------------------------------------------------------------- /hparams/tau19/cassle_simclr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/cassle_simclr.yaml -------------------------------------------------------------------------------- /hparams/tau19/knn_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/knn_eval.yaml -------------------------------------------------------------------------------- /hparams/tau19/knn_eval_10task.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/knn_eval_10task.yaml -------------------------------------------------------------------------------- /hparams/tau19/knn_eval_moco-Copy1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/knn_eval_moco-Copy1.yaml -------------------------------------------------------------------------------- /hparams/tau19/knn_eval_moco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/knn_eval_moco.yaml -------------------------------------------------------------------------------- /hparams/tau19/linclf_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/linclf_train.yaml -------------------------------------------------------------------------------- /hparams/tau19/linclf_train_10task.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/linclf_train_10task.yaml -------------------------------------------------------------------------------- /hparams/tau19/linclf_train_moco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/linclf_train_moco.yaml -------------------------------------------------------------------------------- /hparams/tau19/supBarlowTwins_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/supBarlowTwins_train.yaml -------------------------------------------------------------------------------- /hparams/tau19/sup_labeldist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/sup_labeldist.yaml -------------------------------------------------------------------------------- /hparams/tau19/sup_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/sup_train.yaml -------------------------------------------------------------------------------- /hparams/tau19/supclr_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/supclr_train.yaml -------------------------------------------------------------------------------- /hparams/tau19/supclr_train_10task.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/supclr_train_10task.yaml -------------------------------------------------------------------------------- /hparams/tau19/supmoco_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/tau19/supmoco_train.yaml -------------------------------------------------------------------------------- /hparams/us8k/cassiam_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/cassiam_train.yaml -------------------------------------------------------------------------------- /hparams/us8k/cassle_BarlowTwins.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/cassle_BarlowTwins.yaml -------------------------------------------------------------------------------- /hparams/us8k/cassle_simclr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/cassle_simclr.yaml -------------------------------------------------------------------------------- /hparams/us8k/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/debug.yaml -------------------------------------------------------------------------------- /hparams/us8k/knn_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/knn_eval.yaml -------------------------------------------------------------------------------- /hparams/us8k/knn_eval_10task.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/knn_eval_10task.yaml -------------------------------------------------------------------------------- /hparams/us8k/knn_eval_moco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/knn_eval_moco.yaml -------------------------------------------------------------------------------- /hparams/us8k/linclf_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/linclf_train.yaml -------------------------------------------------------------------------------- /hparams/us8k/linclf_train_10task.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/linclf_train_10task.yaml -------------------------------------------------------------------------------- /hparams/us8k/linclf_train_moco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/linclf_train_moco.yaml -------------------------------------------------------------------------------- /hparams/us8k/moco_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/moco_train.yaml -------------------------------------------------------------------------------- /hparams/us8k/simclr_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/simclr_train.yaml -------------------------------------------------------------------------------- /hparams/us8k/simsiam_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/simsiam_train.yaml -------------------------------------------------------------------------------- /hparams/us8k/supBarlowTwins_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/supBarlowTwins_train.yaml -------------------------------------------------------------------------------- /hparams/us8k/sup_labeldist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/sup_labeldist.yaml -------------------------------------------------------------------------------- /hparams/us8k/sup_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/sup_train.yaml -------------------------------------------------------------------------------- /hparams/us8k/supclr_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/supclr_train.yaml -------------------------------------------------------------------------------- /hparams/us8k/supclr_train_10task.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/supclr_train_10task.yaml -------------------------------------------------------------------------------- /hparams/us8k/supclr_train_bak.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/supclr_train_bak.yaml -------------------------------------------------------------------------------- /hparams/us8k/supmoco_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/supmoco_train.yaml -------------------------------------------------------------------------------- /hparams/us8k/supsiam_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/supsiam_train.yaml -------------------------------------------------------------------------------- /hparams/us8k/train_ecapa_tdnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/train_ecapa_tdnn.yaml -------------------------------------------------------------------------------- /hparams/us8k/visualize.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/us8k/visualize.yaml -------------------------------------------------------------------------------- /hparams/vgg/linclf_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/vgg/linclf_train.yaml -------------------------------------------------------------------------------- /hparams/vgg/supBarlowTwins_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/vgg/supBarlowTwins_train.yaml -------------------------------------------------------------------------------- /hparams/vgg/sup_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/vgg/sup_train.yaml -------------------------------------------------------------------------------- /hparams/vgg/supclr_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/vgg/supclr_train.yaml -------------------------------------------------------------------------------- /hparams/vgg/supmoco_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/hparams/vgg/supmoco_train.yaml -------------------------------------------------------------------------------- /knn_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/knn_eval.py -------------------------------------------------------------------------------- /linclf_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/linclf_train.py -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/losses.py -------------------------------------------------------------------------------- /moco_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/moco_train.py -------------------------------------------------------------------------------- /models/frontend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/models/frontend.py -------------------------------------------------------------------------------- /models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/models/modules.py -------------------------------------------------------------------------------- /models/modules_moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/models/modules_moco.py -------------------------------------------------------------------------------- /models/pann.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/models/pann.py -------------------------------------------------------------------------------- /models/pann_moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/models/pann_moco.py -------------------------------------------------------------------------------- /models/tdnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/models/tdnn.py -------------------------------------------------------------------------------- /schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/schedulers.py -------------------------------------------------------------------------------- /simclr_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/simclr_train.py -------------------------------------------------------------------------------- /simsiam_debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/simsiam_debug.py -------------------------------------------------------------------------------- /simsiam_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/simsiam_train.py -------------------------------------------------------------------------------- /supBarlowTwins_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/supBarlowTwins_train.py -------------------------------------------------------------------------------- /sup_labeldist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/sup_labeldist.py -------------------------------------------------------------------------------- /sup_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/sup_train.py -------------------------------------------------------------------------------- /supclr_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/supclr_train.py -------------------------------------------------------------------------------- /supclr_train_agem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/supclr_train_agem.py -------------------------------------------------------------------------------- /supclr_train_bak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/supclr_train_bak.py -------------------------------------------------------------------------------- /supmoco_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/supmoco_train.py -------------------------------------------------------------------------------- /supsiam_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/supsiam_train.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/utils.py -------------------------------------------------------------------------------- /visualize_embedding.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhepeiw/cssl_sound/HEAD/visualize_embedding.ipynb --------------------------------------------------------------------------------