├── .gitignore ├── LICENSE ├── README.md ├── analysis ├── README.md ├── adversarial_linf.py ├── mine.py ├── train_vaes.py ├── vae.py ├── vae_mi.py └── vgg.py ├── datasets ├── __init__.py ├── bengali.py ├── datasets.py ├── fashion.py ├── google_commands │ ├── README.md │ ├── google_commands.py │ ├── sft_transforms.py │ └── transforms.py ├── imagenet_a.py ├── imagenet_hdf5.py ├── tiny_imagenet.py ├── toxic.py └── toxic_bert.py ├── experiments ├── bengali_experiment.sh ├── cifar_experiment.sh ├── fashion_experiments.sh ├── google_commands_experiment.sh ├── imagenet_experiment.sh ├── modelnet_experiment.sh ├── tiny_imagenet_experiment.sh └── toxic_experiment.sh ├── fmix.py ├── fmix_3d.gif ├── fmix_example.png ├── hubconf.py ├── implementations ├── lightning.py ├── tensorflow_implementation.py ├── test_lightning.py ├── test_tensorflow.py ├── test_torchbearer.py └── torchbearer_implementation.py ├── models ├── __init__.py ├── bert.py ├── densenet3.py ├── models.py ├── pyramid.py ├── resnet.py ├── senet.py ├── toxic_cnn.py ├── toxic_lstm.py └── wide_resnet.py ├── notebooks ├── example_masks.ipynb └── grad_cam.ipynb ├── requirements.txt ├── trainer.py └── utils ├── __init__.py ├── auto_augment ├── __init__.py ├── auto_augment.py └── auto_augment_aug_list.py ├── bengali_evaluate.py ├── convert_imagenet_model.py ├── cross_val.py ├── imagenet_to_hdf5.py ├── lr_warmup.py ├── macro_recall.py ├── msda_alternator.py ├── process.py ├── reduced_dataset_splitter.py └── reformulated_mixup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/README.md -------------------------------------------------------------------------------- /analysis/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/analysis/README.md -------------------------------------------------------------------------------- /analysis/adversarial_linf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/analysis/adversarial_linf.py -------------------------------------------------------------------------------- /analysis/mine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/analysis/mine.py -------------------------------------------------------------------------------- /analysis/train_vaes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/analysis/train_vaes.py -------------------------------------------------------------------------------- /analysis/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/analysis/vae.py -------------------------------------------------------------------------------- /analysis/vae_mi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/analysis/vae_mi.py -------------------------------------------------------------------------------- /analysis/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/analysis/vgg.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/bengali.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/bengali.py -------------------------------------------------------------------------------- /datasets/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/datasets.py -------------------------------------------------------------------------------- /datasets/fashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/fashion.py -------------------------------------------------------------------------------- /datasets/google_commands/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/google_commands/README.md -------------------------------------------------------------------------------- /datasets/google_commands/google_commands.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/google_commands/google_commands.py -------------------------------------------------------------------------------- /datasets/google_commands/sft_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/google_commands/sft_transforms.py -------------------------------------------------------------------------------- /datasets/google_commands/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/google_commands/transforms.py -------------------------------------------------------------------------------- /datasets/imagenet_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/imagenet_a.py -------------------------------------------------------------------------------- /datasets/imagenet_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/imagenet_hdf5.py -------------------------------------------------------------------------------- /datasets/tiny_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/tiny_imagenet.py -------------------------------------------------------------------------------- /datasets/toxic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/toxic.py -------------------------------------------------------------------------------- /datasets/toxic_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/datasets/toxic_bert.py -------------------------------------------------------------------------------- /experiments/bengali_experiment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/experiments/bengali_experiment.sh -------------------------------------------------------------------------------- /experiments/cifar_experiment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/experiments/cifar_experiment.sh -------------------------------------------------------------------------------- /experiments/fashion_experiments.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/experiments/fashion_experiments.sh -------------------------------------------------------------------------------- /experiments/google_commands_experiment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/experiments/google_commands_experiment.sh -------------------------------------------------------------------------------- /experiments/imagenet_experiment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/experiments/imagenet_experiment.sh -------------------------------------------------------------------------------- /experiments/modelnet_experiment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/experiments/modelnet_experiment.sh -------------------------------------------------------------------------------- /experiments/tiny_imagenet_experiment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/experiments/tiny_imagenet_experiment.sh -------------------------------------------------------------------------------- /experiments/toxic_experiment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/experiments/toxic_experiment.sh -------------------------------------------------------------------------------- /fmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/fmix.py -------------------------------------------------------------------------------- /fmix_3d.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/fmix_3d.gif -------------------------------------------------------------------------------- /fmix_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/fmix_example.png -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/hubconf.py -------------------------------------------------------------------------------- /implementations/lightning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/implementations/lightning.py -------------------------------------------------------------------------------- /implementations/tensorflow_implementation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/implementations/tensorflow_implementation.py -------------------------------------------------------------------------------- /implementations/test_lightning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/implementations/test_lightning.py -------------------------------------------------------------------------------- /implementations/test_tensorflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/implementations/test_tensorflow.py -------------------------------------------------------------------------------- /implementations/test_torchbearer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/implementations/test_torchbearer.py -------------------------------------------------------------------------------- /implementations/torchbearer_implementation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/implementations/torchbearer_implementation.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/models/bert.py -------------------------------------------------------------------------------- /models/densenet3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/models/densenet3.py -------------------------------------------------------------------------------- /models/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/models/models.py -------------------------------------------------------------------------------- /models/pyramid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/models/pyramid.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/models/senet.py -------------------------------------------------------------------------------- /models/toxic_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/models/toxic_cnn.py -------------------------------------------------------------------------------- /models/toxic_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/models/toxic_lstm.py -------------------------------------------------------------------------------- /models/wide_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/models/wide_resnet.py -------------------------------------------------------------------------------- /notebooks/example_masks.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/notebooks/example_masks.ipynb -------------------------------------------------------------------------------- /notebooks/grad_cam.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/notebooks/grad_cam.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/requirements.txt -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/auto_augment/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/auto_augment/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/auto_augment/auto_augment.py -------------------------------------------------------------------------------- /utils/auto_augment/auto_augment_aug_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/auto_augment/auto_augment_aug_list.py -------------------------------------------------------------------------------- /utils/bengali_evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/bengali_evaluate.py -------------------------------------------------------------------------------- /utils/convert_imagenet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/convert_imagenet_model.py -------------------------------------------------------------------------------- /utils/cross_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/cross_val.py -------------------------------------------------------------------------------- /utils/imagenet_to_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/imagenet_to_hdf5.py -------------------------------------------------------------------------------- /utils/lr_warmup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/lr_warmup.py -------------------------------------------------------------------------------- /utils/macro_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/macro_recall.py -------------------------------------------------------------------------------- /utils/msda_alternator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/msda_alternator.py -------------------------------------------------------------------------------- /utils/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/process.py -------------------------------------------------------------------------------- /utils/reduced_dataset_splitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/reduced_dataset_splitter.py -------------------------------------------------------------------------------- /utils/reformulated_mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecs-vlc/FMix/HEAD/utils/reformulated_mixup.py --------------------------------------------------------------------------------