├── .gitignore ├── LICENSE ├── README.md ├── cfgs ├── crnn_chunks_melspec.cfg ├── densenet121_chunks_melspec.cfg ├── resnet18_chunks.cfg ├── resnet18_chunks_melspec.cfg ├── vgglike_chunks.cfg └── vgglike_chunks_melspec.cfg ├── chunkify_fsd50k.py ├── eval_fsd50k.py ├── make_chunks_manifests.py ├── multiproc_resample.py ├── readme_images └── specs_wide2.png ├── requirements.txt ├── src ├── __init__.py ├── data │ ├── __init__.py │ ├── audio_parser.py │ ├── dataset.py │ ├── fsd_eval_dataset.py │ ├── mixers.py │ ├── random_resized_crop.py │ ├── transforms.py │ └── utils.py ├── models │ ├── __init__.py │ ├── crnn.py │ ├── densenet.py │ ├── fsd50k_lightning.py │ ├── netvlad.py │ ├── resnet.py │ └── vgglike.py ├── optim │ ├── __init__.py │ └── focal_loss.py └── utilities │ ├── __init__.py │ ├── config_parser.py │ └── metrics_helper.py └── train_fsd50k_chunks.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /cfgs/crnn_chunks_melspec.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/cfgs/crnn_chunks_melspec.cfg -------------------------------------------------------------------------------- /cfgs/densenet121_chunks_melspec.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/cfgs/densenet121_chunks_melspec.cfg -------------------------------------------------------------------------------- /cfgs/resnet18_chunks.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/cfgs/resnet18_chunks.cfg -------------------------------------------------------------------------------- /cfgs/resnet18_chunks_melspec.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/cfgs/resnet18_chunks_melspec.cfg -------------------------------------------------------------------------------- /cfgs/vgglike_chunks.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/cfgs/vgglike_chunks.cfg -------------------------------------------------------------------------------- /cfgs/vgglike_chunks_melspec.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/cfgs/vgglike_chunks_melspec.cfg -------------------------------------------------------------------------------- /chunkify_fsd50k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/chunkify_fsd50k.py -------------------------------------------------------------------------------- /eval_fsd50k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/eval_fsd50k.py -------------------------------------------------------------------------------- /make_chunks_manifests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/make_chunks_manifests.py -------------------------------------------------------------------------------- /multiproc_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/multiproc_resample.py -------------------------------------------------------------------------------- /readme_images/specs_wide2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/readme_images/specs_wide2.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/audio_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/data/audio_parser.py -------------------------------------------------------------------------------- /src/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/data/dataset.py -------------------------------------------------------------------------------- /src/data/fsd_eval_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/data/fsd_eval_dataset.py -------------------------------------------------------------------------------- /src/data/mixers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/data/mixers.py -------------------------------------------------------------------------------- /src/data/random_resized_crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/data/random_resized_crop.py -------------------------------------------------------------------------------- /src/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/data/transforms.py -------------------------------------------------------------------------------- /src/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/data/utils.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/crnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/models/crnn.py -------------------------------------------------------------------------------- /src/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/models/densenet.py -------------------------------------------------------------------------------- /src/models/fsd50k_lightning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/models/fsd50k_lightning.py -------------------------------------------------------------------------------- /src/models/netvlad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/models/netvlad.py -------------------------------------------------------------------------------- /src/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/models/resnet.py -------------------------------------------------------------------------------- /src/models/vgglike.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/models/vgglike.py -------------------------------------------------------------------------------- /src/optim/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/optim/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/optim/focal_loss.py -------------------------------------------------------------------------------- /src/utilities/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utilities/config_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/utilities/config_parser.py -------------------------------------------------------------------------------- /src/utilities/metrics_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/src/utilities/metrics_helper.py -------------------------------------------------------------------------------- /train_fsd50k_chunks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/fsd50k-pytorch/HEAD/train_fsd50k_chunks.py --------------------------------------------------------------------------------