├── .dockerignore ├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── colab_demo.ipynb ├── easy_ViTPose ├── __init__.py ├── config.yaml ├── configs │ ├── ViTPose_aic.py │ ├── ViTPose_ap10k.py │ ├── ViTPose_apt36k.py │ ├── ViTPose_coco.py │ ├── ViTPose_coco_25.py │ ├── ViTPose_common.py │ ├── ViTPose_custom.py │ ├── ViTPose_mpii.py │ ├── ViTPose_wholebody.py │ ├── __init__.py │ └── train_configs │ │ ├── ViTPose_base_coco_256x192.py │ │ ├── ViTPose_base_simple_coco_256x192.py │ │ ├── ViTPose_huge_coco_256x192.py │ │ ├── ViTPose_huge_simple_coco_256x192.py │ │ ├── ViTPose_large_coco_256x192.py │ │ ├── ViTPose_large_coco_256x192_custom.py │ │ └── ViTPose_large_simple_coco_256x192.py ├── datasets │ ├── COCO.py │ ├── HumanPoseEstimation.py │ └── __init__.py ├── easy_ViTPose.egg-info │ ├── PKG-INFO │ ├── SOURCES.txt │ ├── dependency_links.txt │ └── top_level.txt ├── inference.py ├── sort.py ├── to_onnx.ipynb ├── to_trt.ipynb ├── train.py ├── vit_models │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ └── vit.py │ ├── head │ │ ├── __init__.py │ │ ├── topdown_heatmap_base_head.py │ │ └── topdown_heatmap_simple_head.py │ ├── losses │ │ ├── __init__.py │ │ ├── classfication_loss.py │ │ ├── heatmap_loss.py │ │ ├── mesh_loss.py │ │ ├── mse_loss.py │ │ ├── multi_loss_factory.py │ │ └── regression_loss.py │ ├── model.py │ └── optimizer.py └── vit_utils │ ├── __init__.py │ ├── dist_util.py │ ├── inference.py │ ├── logging.py │ ├── nms │ ├── __init__.py │ ├── cpu_nms.c │ ├── cpu_nms.cpython-37m-x86_64-linux-gnu.so │ ├── cpu_nms.cpython-39-x86_64-linux-gnu.so │ ├── cpu_nms.pyx │ ├── gpu_nms.cpp │ ├── gpu_nms.cpython-37m-x86_64-linux-gnu.so │ ├── gpu_nms.cpython-39-x86_64-linux-gnu.so │ ├── gpu_nms.cu │ ├── gpu_nms.hpp │ ├── gpu_nms.pyx │ ├── nms.py │ ├── nms_kernel.cu │ ├── nms_ori.py │ └── setup_linux.py │ ├── post_processing │ ├── __init__.py │ ├── group.py │ ├── nms.py │ ├── one_euro_filter.py │ └── post_transforms.py │ ├── top_down_eval.py │ ├── train_valid_fn.py │ ├── transform.py │ ├── util.py │ └── visualization.py ├── evaluation_on_coco.py ├── export.py ├── inference.py ├── model_split.py ├── models ├── .gitignore └── download.sh ├── requirements.txt ├── requirements_gpu.txt └── setup.py /.dockerignore: -------------------------------------------------------------------------------- 1 | Dockerfile 2 | models 3 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/README.md -------------------------------------------------------------------------------- /colab_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/colab_demo.ipynb -------------------------------------------------------------------------------- /easy_ViTPose/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/__init__.py -------------------------------------------------------------------------------- /easy_ViTPose/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/config.yaml -------------------------------------------------------------------------------- /easy_ViTPose/configs/ViTPose_aic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/ViTPose_aic.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/ViTPose_ap10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/ViTPose_ap10k.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/ViTPose_apt36k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/ViTPose_apt36k.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/ViTPose_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/ViTPose_coco.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/ViTPose_coco_25.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/ViTPose_coco_25.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/ViTPose_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/ViTPose_common.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/ViTPose_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/ViTPose_custom.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/ViTPose_mpii.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/ViTPose_mpii.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/ViTPose_wholebody.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/ViTPose_wholebody.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /easy_ViTPose/configs/train_configs/ViTPose_base_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/train_configs/ViTPose_base_coco_256x192.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/train_configs/ViTPose_base_simple_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/train_configs/ViTPose_base_simple_coco_256x192.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/train_configs/ViTPose_huge_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/train_configs/ViTPose_huge_coco_256x192.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/train_configs/ViTPose_huge_simple_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/train_configs/ViTPose_huge_simple_coco_256x192.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/train_configs/ViTPose_large_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/train_configs/ViTPose_large_coco_256x192.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/train_configs/ViTPose_large_coco_256x192_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/train_configs/ViTPose_large_coco_256x192_custom.py -------------------------------------------------------------------------------- /easy_ViTPose/configs/train_configs/ViTPose_large_simple_coco_256x192.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/configs/train_configs/ViTPose_large_simple_coco_256x192.py -------------------------------------------------------------------------------- /easy_ViTPose/datasets/COCO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/datasets/COCO.py -------------------------------------------------------------------------------- /easy_ViTPose/datasets/HumanPoseEstimation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/datasets/HumanPoseEstimation.py -------------------------------------------------------------------------------- /easy_ViTPose/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /easy_ViTPose/easy_ViTPose.egg-info/PKG-INFO: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/easy_ViTPose.egg-info/PKG-INFO -------------------------------------------------------------------------------- /easy_ViTPose/easy_ViTPose.egg-info/SOURCES.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/easy_ViTPose.egg-info/SOURCES.txt -------------------------------------------------------------------------------- /easy_ViTPose/easy_ViTPose.egg-info/dependency_links.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /easy_ViTPose/easy_ViTPose.egg-info/top_level.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/easy_ViTPose.egg-info/top_level.txt -------------------------------------------------------------------------------- /easy_ViTPose/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/inference.py -------------------------------------------------------------------------------- /easy_ViTPose/sort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/sort.py -------------------------------------------------------------------------------- /easy_ViTPose/to_onnx.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/to_onnx.ipynb -------------------------------------------------------------------------------- /easy_ViTPose/to_trt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/to_trt.ipynb -------------------------------------------------------------------------------- /easy_ViTPose/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/train.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/__init__.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/backbone/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/backbone/vit.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/head/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/head/topdown_heatmap_base_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/head/topdown_heatmap_base_head.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/head/topdown_heatmap_simple_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/head/topdown_heatmap_simple_head.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/losses/__init__.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/losses/classfication_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/losses/classfication_loss.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/losses/heatmap_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/losses/heatmap_loss.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/losses/mesh_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/losses/mesh_loss.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/losses/mse_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/losses/mse_loss.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/losses/multi_loss_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/losses/multi_loss_factory.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/losses/regression_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/losses/regression_loss.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/model.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_models/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_models/optimizer.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/__init__.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/dist_util.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/inference.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/logging.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/cpu_nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/cpu_nms.c -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/cpu_nms.cpython-37m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/cpu_nms.cpython-37m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/cpu_nms.cpython-39-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/cpu_nms.cpython-39-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/cpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/cpu_nms.pyx -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/gpu_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/gpu_nms.cpp -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/gpu_nms.cpython-37m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/gpu_nms.cpython-37m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/gpu_nms.cpython-39-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/gpu_nms.cpython-39-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/gpu_nms.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/gpu_nms.cu -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/gpu_nms.hpp -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/gpu_nms.pyx -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/nms.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/nms_kernel.cu -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/nms_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/nms_ori.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/nms/setup_linux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/nms/setup_linux.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/post_processing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/post_processing/__init__.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/post_processing/group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/post_processing/group.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/post_processing/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/post_processing/nms.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/post_processing/one_euro_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/post_processing/one_euro_filter.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/post_processing/post_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/post_processing/post_transforms.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/top_down_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/top_down_eval.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/train_valid_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/train_valid_fn.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/transform.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/util.py -------------------------------------------------------------------------------- /easy_ViTPose/vit_utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/easy_ViTPose/vit_utils/visualization.py -------------------------------------------------------------------------------- /evaluation_on_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/evaluation_on_coco.py -------------------------------------------------------------------------------- /export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/export.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/inference.py -------------------------------------------------------------------------------- /model_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/model_split.py -------------------------------------------------------------------------------- /models/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/models/.gitignore -------------------------------------------------------------------------------- /models/download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/models/download.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements_gpu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/requirements_gpu.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JunkyByte/easy_ViTPose/HEAD/setup.py --------------------------------------------------------------------------------