├── .gitignore ├── LICENSE ├── README.md ├── data_preprocessing └── match_coco_cond.py ├── experiments ├── coco │ └── hrnet │ │ ├── w32_384x288_adam_lr1e-3.yaml │ │ └── w48_384x288_adam_lr1e-3.yaml └── crowdpose │ └── hrnet │ ├── w32_384x288_adam_lr1e-3.yaml │ └── w48_384x288_adam_lr1e-3.yaml ├── lib ├── Makefile ├── analysis │ ├── evaluation.py │ └── qualitative_evaluation.py ├── config │ ├── __init__.py │ ├── default.py │ └── models.py ├── core │ ├── evaluate.py │ ├── function.py │ ├── inference.py │ ├── loss.py │ ├── train.py │ └── validate.py ├── dataset │ ├── JointsDataset.py │ ├── __init__.py │ ├── coco.py │ ├── crowdpose.py │ ├── dataloader.py │ ├── fish.py │ ├── marmosets.py │ ├── multimouse.py │ ├── ochuman.py │ └── pose_synthesis.py ├── models │ ├── __init__.py │ ├── pose_hrnet.py │ ├── pose_hrnet_coam.py │ ├── pose_resnet.py │ ├── self_attention.py │ └── transpose_h.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 │ ├── gaussian.py │ ├── transforms.py │ ├── utils.py │ ├── vis.py │ ├── vis_coco.py │ └── zipreader.py ├── media ├── 000000.jpg └── BUCTD_fig1.png ├── requirements.txt ├── scripts ├── test │ ├── test_BUCTD_COAM_gen_sample.sh │ └── test_BUCTD_prenet_gen_sample.sh └── train │ ├── train_BUCTD_COAM.sh │ ├── train_BUCTD_COAM_gen_sample.sh │ ├── train_BUCTD_prenet_gen_sample.sh │ └── train_BUCTD_transpose.sh └── tools ├── _init_paths.py ├── inference.py ├── test.py ├── train.py └── vis.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/README.md -------------------------------------------------------------------------------- /data_preprocessing/match_coco_cond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/data_preprocessing/match_coco_cond.py -------------------------------------------------------------------------------- /experiments/coco/hrnet/w32_384x288_adam_lr1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/experiments/coco/hrnet/w32_384x288_adam_lr1e-3.yaml -------------------------------------------------------------------------------- /experiments/coco/hrnet/w48_384x288_adam_lr1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/experiments/coco/hrnet/w48_384x288_adam_lr1e-3.yaml -------------------------------------------------------------------------------- /experiments/crowdpose/hrnet/w32_384x288_adam_lr1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/experiments/crowdpose/hrnet/w32_384x288_adam_lr1e-3.yaml -------------------------------------------------------------------------------- /experiments/crowdpose/hrnet/w48_384x288_adam_lr1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/experiments/crowdpose/hrnet/w48_384x288_adam_lr1e-3.yaml -------------------------------------------------------------------------------- /lib/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/Makefile -------------------------------------------------------------------------------- /lib/analysis/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/analysis/evaluation.py -------------------------------------------------------------------------------- /lib/analysis/qualitative_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/analysis/qualitative_evaluation.py -------------------------------------------------------------------------------- /lib/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/config/__init__.py -------------------------------------------------------------------------------- /lib/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/config/default.py -------------------------------------------------------------------------------- /lib/config/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/config/models.py -------------------------------------------------------------------------------- /lib/core/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/core/evaluate.py -------------------------------------------------------------------------------- /lib/core/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/core/function.py -------------------------------------------------------------------------------- /lib/core/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/core/inference.py -------------------------------------------------------------------------------- /lib/core/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/core/loss.py -------------------------------------------------------------------------------- /lib/core/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/core/train.py -------------------------------------------------------------------------------- /lib/core/validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/core/validate.py -------------------------------------------------------------------------------- /lib/dataset/JointsDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/dataset/JointsDataset.py -------------------------------------------------------------------------------- /lib/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/dataset/__init__.py -------------------------------------------------------------------------------- /lib/dataset/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/dataset/coco.py -------------------------------------------------------------------------------- /lib/dataset/crowdpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/dataset/crowdpose.py -------------------------------------------------------------------------------- /lib/dataset/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/dataset/dataloader.py -------------------------------------------------------------------------------- /lib/dataset/fish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/dataset/fish.py -------------------------------------------------------------------------------- /lib/dataset/marmosets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/dataset/marmosets.py -------------------------------------------------------------------------------- /lib/dataset/multimouse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/dataset/multimouse.py -------------------------------------------------------------------------------- /lib/dataset/ochuman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/dataset/ochuman.py -------------------------------------------------------------------------------- /lib/dataset/pose_synthesis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/dataset/pose_synthesis.py -------------------------------------------------------------------------------- /lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/models/__init__.py -------------------------------------------------------------------------------- /lib/models/pose_hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/models/pose_hrnet.py -------------------------------------------------------------------------------- /lib/models/pose_hrnet_coam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/models/pose_hrnet_coam.py -------------------------------------------------------------------------------- /lib/models/pose_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/models/pose_resnet.py -------------------------------------------------------------------------------- /lib/models/self_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/models/self_attention.py -------------------------------------------------------------------------------- /lib/models/transpose_h.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/models/transpose_h.py -------------------------------------------------------------------------------- /lib/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/nms/cpu_nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/nms/cpu_nms.c -------------------------------------------------------------------------------- /lib/nms/cpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/nms/cpu_nms.pyx -------------------------------------------------------------------------------- /lib/nms/gpu_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/nms/gpu_nms.cpp -------------------------------------------------------------------------------- /lib/nms/gpu_nms.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/nms/gpu_nms.cu -------------------------------------------------------------------------------- /lib/nms/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/nms/gpu_nms.hpp -------------------------------------------------------------------------------- /lib/nms/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/nms/gpu_nms.pyx -------------------------------------------------------------------------------- /lib/nms/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/nms/nms.py -------------------------------------------------------------------------------- /lib/nms/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/nms/nms_kernel.cu -------------------------------------------------------------------------------- /lib/nms/setup_linux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/nms/setup_linux.py -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/utils/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/utils/gaussian.py -------------------------------------------------------------------------------- /lib/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/utils/transforms.py -------------------------------------------------------------------------------- /lib/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/utils/utils.py -------------------------------------------------------------------------------- /lib/utils/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/utils/vis.py -------------------------------------------------------------------------------- /lib/utils/vis_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/utils/vis_coco.py -------------------------------------------------------------------------------- /lib/utils/zipreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/lib/utils/zipreader.py -------------------------------------------------------------------------------- /media/000000.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/media/000000.jpg -------------------------------------------------------------------------------- /media/BUCTD_fig1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/media/BUCTD_fig1.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/test/test_BUCTD_COAM_gen_sample.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/scripts/test/test_BUCTD_COAM_gen_sample.sh -------------------------------------------------------------------------------- /scripts/test/test_BUCTD_prenet_gen_sample.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/scripts/test/test_BUCTD_prenet_gen_sample.sh -------------------------------------------------------------------------------- /scripts/train/train_BUCTD_COAM.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/scripts/train/train_BUCTD_COAM.sh -------------------------------------------------------------------------------- /scripts/train/train_BUCTD_COAM_gen_sample.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/scripts/train/train_BUCTD_COAM_gen_sample.sh -------------------------------------------------------------------------------- /scripts/train/train_BUCTD_prenet_gen_sample.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/scripts/train/train_BUCTD_prenet_gen_sample.sh -------------------------------------------------------------------------------- /scripts/train/train_BUCTD_transpose.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/scripts/train/train_BUCTD_transpose.sh -------------------------------------------------------------------------------- /tools/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/tools/_init_paths.py -------------------------------------------------------------------------------- /tools/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/tools/inference.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amathislab/BUCTD/HEAD/tools/vis.py --------------------------------------------------------------------------------