├── .gitattributes ├── README.md ├── config ├── magna │ ├── east_final_openl3.json │ ├── east_final_pann.json │ ├── east_final_passt.json │ ├── east_final_vggish.json │ ├── east_kd_openl3.json │ ├── east_kd_pann.json │ ├── east_kd_passt.json │ ├── east_kd_vggish.json │ ├── lr_openl3.json │ ├── lr_pann.json │ ├── lr_passt.json │ └── lr_vggish.json └── openmic │ ├── east_final_openl3.json │ ├── east_final_pann.json │ ├── east_final_passt.json │ ├── east_final_vggish.json │ ├── east_kd_openl3.json │ ├── east_kd_pann.json │ ├── east_kd_passt.json │ ├── east_kd_vggish.json │ ├── lr_openl3.json │ ├── lr_pann.json │ ├── lr_passt.json │ └── lr_vggish.json ├── data ├── __init__.py ├── datamodule.py ├── dataset.py └── featurizer │ ├── build_featurizer.py │ ├── log_mel_featurizer.py │ └── waveform_featurizer.py ├── eval_model.py ├── magna_test.json ├── magna_train.json ├── magna_valid.json ├── models ├── __init__.py ├── basemodels │ ├── __init__.py │ ├── cp_resnet.py │ ├── mobile_fcn.py │ └── pretrained_feature_classifier.py ├── metrics │ ├── __init__.py │ └── metrics.py └── reg_loss_fn │ ├── __init__.py │ ├── combine_dist_reg.py │ ├── feature_space_reg.py │ └── kd.py ├── openmic_test.json ├── openmic_train.json ├── openmic_valid.json ├── preprocess ├── parse_magna.py ├── parse_openmic.py ├── pretrained_feature_extractor │ ├── extract_openl3.py │ ├── extract_passt.py │ └── extract_vggish.py ├── resample_openmic.py ├── split_magna.py └── split_openmic_train.py ├── train.py ├── work_dir_magna ├── east_final_openl3 │ └── epoch=93-val_mAP=0.448.ckpt ├── east_final_pann │ └── epoch=82-val_mAP=0.450.ckpt ├── east_final_passt │ └── epoch=64-val_mAP=0.466.ckpt └── east_final_vggish │ └── epoch=67-val_mAP=0.451.ckpt └── work_dir_openmic ├── east_final_openl3 └── epoch=62-val_mAP=0.851.ckpt ├── east_final_pann └── epoch=90-val_mAP=0.860.ckpt ├── east_final_passt └── epoch=96-val_mAP=0.864.ckpt └── east_final_vggish └── epoch=75-val_mAP=0.854.ckpt /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/.gitattributes -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/README.md -------------------------------------------------------------------------------- /config/magna/east_final_openl3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/east_final_openl3.json -------------------------------------------------------------------------------- /config/magna/east_final_pann.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/east_final_pann.json -------------------------------------------------------------------------------- /config/magna/east_final_passt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/east_final_passt.json -------------------------------------------------------------------------------- /config/magna/east_final_vggish.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/east_final_vggish.json -------------------------------------------------------------------------------- /config/magna/east_kd_openl3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/east_kd_openl3.json -------------------------------------------------------------------------------- /config/magna/east_kd_pann.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/east_kd_pann.json -------------------------------------------------------------------------------- /config/magna/east_kd_passt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/east_kd_passt.json -------------------------------------------------------------------------------- /config/magna/east_kd_vggish.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/east_kd_vggish.json -------------------------------------------------------------------------------- /config/magna/lr_openl3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/lr_openl3.json -------------------------------------------------------------------------------- /config/magna/lr_pann.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/lr_pann.json -------------------------------------------------------------------------------- /config/magna/lr_passt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/lr_passt.json -------------------------------------------------------------------------------- /config/magna/lr_vggish.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/magna/lr_vggish.json -------------------------------------------------------------------------------- /config/openmic/east_final_openl3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/east_final_openl3.json -------------------------------------------------------------------------------- /config/openmic/east_final_pann.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/east_final_pann.json -------------------------------------------------------------------------------- /config/openmic/east_final_passt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/east_final_passt.json -------------------------------------------------------------------------------- /config/openmic/east_final_vggish.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/east_final_vggish.json -------------------------------------------------------------------------------- /config/openmic/east_kd_openl3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/east_kd_openl3.json -------------------------------------------------------------------------------- /config/openmic/east_kd_pann.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/east_kd_pann.json -------------------------------------------------------------------------------- /config/openmic/east_kd_passt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/east_kd_passt.json -------------------------------------------------------------------------------- /config/openmic/east_kd_vggish.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/east_kd_vggish.json -------------------------------------------------------------------------------- /config/openmic/lr_openl3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/lr_openl3.json -------------------------------------------------------------------------------- /config/openmic/lr_pann.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/lr_pann.json -------------------------------------------------------------------------------- /config/openmic/lr_passt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/lr_passt.json -------------------------------------------------------------------------------- /config/openmic/lr_vggish.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/config/openmic/lr_vggish.json -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/data/datamodule.py -------------------------------------------------------------------------------- /data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/data/dataset.py -------------------------------------------------------------------------------- /data/featurizer/build_featurizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/data/featurizer/build_featurizer.py -------------------------------------------------------------------------------- /data/featurizer/log_mel_featurizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/data/featurizer/log_mel_featurizer.py -------------------------------------------------------------------------------- /data/featurizer/waveform_featurizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/data/featurizer/waveform_featurizer.py -------------------------------------------------------------------------------- /eval_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/eval_model.py -------------------------------------------------------------------------------- /magna_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/magna_test.json -------------------------------------------------------------------------------- /magna_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/magna_train.json -------------------------------------------------------------------------------- /magna_valid.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/magna_valid.json -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/basemodels/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/models/basemodels/__init__.py -------------------------------------------------------------------------------- /models/basemodels/cp_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/models/basemodels/cp_resnet.py -------------------------------------------------------------------------------- /models/basemodels/mobile_fcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/models/basemodels/mobile_fcn.py -------------------------------------------------------------------------------- /models/basemodels/pretrained_feature_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/models/basemodels/pretrained_feature_classifier.py -------------------------------------------------------------------------------- /models/metrics/__init__.py: -------------------------------------------------------------------------------- 1 | from .metrics import MultiLabelBinaryEval -------------------------------------------------------------------------------- /models/metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/models/metrics/metrics.py -------------------------------------------------------------------------------- /models/reg_loss_fn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/models/reg_loss_fn/__init__.py -------------------------------------------------------------------------------- /models/reg_loss_fn/combine_dist_reg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/models/reg_loss_fn/combine_dist_reg.py -------------------------------------------------------------------------------- /models/reg_loss_fn/feature_space_reg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/models/reg_loss_fn/feature_space_reg.py -------------------------------------------------------------------------------- /models/reg_loss_fn/kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/models/reg_loss_fn/kd.py -------------------------------------------------------------------------------- /openmic_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/openmic_test.json -------------------------------------------------------------------------------- /openmic_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/openmic_train.json -------------------------------------------------------------------------------- /openmic_valid.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/openmic_valid.json -------------------------------------------------------------------------------- /preprocess/parse_magna.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/preprocess/parse_magna.py -------------------------------------------------------------------------------- /preprocess/parse_openmic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/preprocess/parse_openmic.py -------------------------------------------------------------------------------- /preprocess/pretrained_feature_extractor/extract_openl3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/preprocess/pretrained_feature_extractor/extract_openl3.py -------------------------------------------------------------------------------- /preprocess/pretrained_feature_extractor/extract_passt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/preprocess/pretrained_feature_extractor/extract_passt.py -------------------------------------------------------------------------------- /preprocess/pretrained_feature_extractor/extract_vggish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/preprocess/pretrained_feature_extractor/extract_vggish.py -------------------------------------------------------------------------------- /preprocess/resample_openmic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/preprocess/resample_openmic.py -------------------------------------------------------------------------------- /preprocess/split_magna.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/preprocess/split_magna.py -------------------------------------------------------------------------------- /preprocess/split_openmic_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/preprocess/split_openmic_train.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/train.py -------------------------------------------------------------------------------- /work_dir_magna/east_final_openl3/epoch=93-val_mAP=0.448.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/work_dir_magna/east_final_openl3/epoch=93-val_mAP=0.448.ckpt -------------------------------------------------------------------------------- /work_dir_magna/east_final_pann/epoch=82-val_mAP=0.450.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/work_dir_magna/east_final_pann/epoch=82-val_mAP=0.450.ckpt -------------------------------------------------------------------------------- /work_dir_magna/east_final_passt/epoch=64-val_mAP=0.466.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/work_dir_magna/east_final_passt/epoch=64-val_mAP=0.466.ckpt -------------------------------------------------------------------------------- /work_dir_magna/east_final_vggish/epoch=67-val_mAP=0.451.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/work_dir_magna/east_final_vggish/epoch=67-val_mAP=0.451.ckpt -------------------------------------------------------------------------------- /work_dir_openmic/east_final_openl3/epoch=62-val_mAP=0.851.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/work_dir_openmic/east_final_openl3/epoch=62-val_mAP=0.851.ckpt -------------------------------------------------------------------------------- /work_dir_openmic/east_final_pann/epoch=90-val_mAP=0.860.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/work_dir_openmic/east_final_pann/epoch=90-val_mAP=0.860.ckpt -------------------------------------------------------------------------------- /work_dir_openmic/east_final_passt/epoch=96-val_mAP=0.864.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/work_dir_openmic/east_final_passt/epoch=96-val_mAP=0.864.ckpt -------------------------------------------------------------------------------- /work_dir_openmic/east_final_vggish/epoch=75-val_mAP=0.854.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suncerock/EAsT-music-classification/HEAD/work_dir_openmic/east_final_vggish/epoch=75-val_mAP=0.854.ckpt --------------------------------------------------------------------------------