├── .gitignore ├── LICENSE ├── README.md ├── experiments ├── coco │ └── resnet │ │ └── res50_256x192_d256x3_adam_lr1e-3_advmix.yaml └── mpii │ ├── hrnet │ └── w32_256x256_adam_lr1e-3_advmix.yaml │ └── resnet │ └── res50_256x256_d256x3_adam_lr1e-3_advmix.yaml ├── figures ├── AdvMix.jpg ├── Qualitative.png ├── benchmarking_results.png └── image_corruption.png ├── lib ├── Makefile ├── config │ ├── __init__.py │ ├── default.py │ └── models.py ├── core │ ├── evaluate.py │ ├── function.py │ ├── inference.py │ └── loss.py ├── dataset │ ├── JointsDataset.py │ ├── __init__.py │ ├── advaug.py │ ├── coco.py │ └── mpii.py ├── models │ ├── Unet_generator.py │ ├── __init__.py │ ├── pose_hrnet.py │ └── pose_resnet.py ├── nms │ ├── __init__.py │ ├── cpu_nms.c │ ├── cpu_nms.pyx │ ├── gpu_nms.cpp │ ├── gpu_nms.cu │ ├── gpu_nms.hpp │ ├── gpu_nms.pyx │ ├── nms.py │ ├── nms_kernel.cu │ └── setup_linux.py └── utils │ ├── __init__.py │ ├── transforms.py │ ├── utils.py │ ├── vis.py │ └── zipreader.py ├── requirements.txt ├── scripts ├── make_datasets.sh ├── test.sh └── train.sh └── tools ├── _init_parse.py ├── _init_paths.py ├── make_datasets.py ├── test_corruption.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/README.md -------------------------------------------------------------------------------- /experiments/coco/resnet/res50_256x192_d256x3_adam_lr1e-3_advmix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/experiments/coco/resnet/res50_256x192_d256x3_adam_lr1e-3_advmix.yaml -------------------------------------------------------------------------------- /experiments/mpii/hrnet/w32_256x256_adam_lr1e-3_advmix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/experiments/mpii/hrnet/w32_256x256_adam_lr1e-3_advmix.yaml -------------------------------------------------------------------------------- /experiments/mpii/resnet/res50_256x256_d256x3_adam_lr1e-3_advmix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/experiments/mpii/resnet/res50_256x256_d256x3_adam_lr1e-3_advmix.yaml -------------------------------------------------------------------------------- /figures/AdvMix.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/figures/AdvMix.jpg -------------------------------------------------------------------------------- /figures/Qualitative.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/figures/Qualitative.png -------------------------------------------------------------------------------- /figures/benchmarking_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/figures/benchmarking_results.png -------------------------------------------------------------------------------- /figures/image_corruption.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/figures/image_corruption.png -------------------------------------------------------------------------------- /lib/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/Makefile -------------------------------------------------------------------------------- /lib/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/config/__init__.py -------------------------------------------------------------------------------- /lib/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/config/default.py -------------------------------------------------------------------------------- /lib/config/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/config/models.py -------------------------------------------------------------------------------- /lib/core/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/core/evaluate.py -------------------------------------------------------------------------------- /lib/core/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/core/function.py -------------------------------------------------------------------------------- /lib/core/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/core/inference.py -------------------------------------------------------------------------------- /lib/core/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/core/loss.py -------------------------------------------------------------------------------- /lib/dataset/JointsDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/dataset/JointsDataset.py -------------------------------------------------------------------------------- /lib/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/dataset/__init__.py -------------------------------------------------------------------------------- /lib/dataset/advaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/dataset/advaug.py -------------------------------------------------------------------------------- /lib/dataset/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/dataset/coco.py -------------------------------------------------------------------------------- /lib/dataset/mpii.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/dataset/mpii.py -------------------------------------------------------------------------------- /lib/models/Unet_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/models/Unet_generator.py -------------------------------------------------------------------------------- /lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/models/__init__.py -------------------------------------------------------------------------------- /lib/models/pose_hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/models/pose_hrnet.py -------------------------------------------------------------------------------- /lib/models/pose_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/models/pose_resnet.py -------------------------------------------------------------------------------- /lib/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/nms/cpu_nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/nms/cpu_nms.c -------------------------------------------------------------------------------- /lib/nms/cpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/nms/cpu_nms.pyx -------------------------------------------------------------------------------- /lib/nms/gpu_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/nms/gpu_nms.cpp -------------------------------------------------------------------------------- /lib/nms/gpu_nms.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/nms/gpu_nms.cu -------------------------------------------------------------------------------- /lib/nms/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/nms/gpu_nms.hpp -------------------------------------------------------------------------------- /lib/nms/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/nms/gpu_nms.pyx -------------------------------------------------------------------------------- /lib/nms/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/nms/nms.py -------------------------------------------------------------------------------- /lib/nms/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/nms/nms_kernel.cu -------------------------------------------------------------------------------- /lib/nms/setup_linux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/nms/setup_linux.py -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/utils/transforms.py -------------------------------------------------------------------------------- /lib/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/utils/utils.py -------------------------------------------------------------------------------- /lib/utils/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/utils/vis.py -------------------------------------------------------------------------------- /lib/utils/zipreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/lib/utils/zipreader.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/make_datasets.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/scripts/make_datasets.sh -------------------------------------------------------------------------------- /scripts/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/scripts/test.sh -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/scripts/train.sh -------------------------------------------------------------------------------- /tools/_init_parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/tools/_init_parse.py -------------------------------------------------------------------------------- /tools/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/tools/_init_paths.py -------------------------------------------------------------------------------- /tools/make_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/tools/make_datasets.py -------------------------------------------------------------------------------- /tools/test_corruption.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/tools/test_corruption.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AIprogrammer/AdvMix/HEAD/tools/train.py --------------------------------------------------------------------------------