├── .gitignore ├── LICENSE ├── README.md ├── callbacks_model.py ├── common.py ├── config ├── dataset │ ├── ai_challenger-colab.cfg │ ├── ai_challenger-gpu.cfg │ ├── ai_challenger-mac.cfg │ ├── coco2017-gpu.cfg │ ├── coco_single_person_only-gpu.cfg │ └── receipt5982-gpu.cfg └── training │ ├── coco_single_experiment01-cpm-sg4-gpu.cfg │ ├── experiment01-colab.cfg │ ├── experiment01-mac.cfg │ ├── experiment01.cfg │ ├── experiment02-cpm-gpu.cfg │ ├── experiment02-cpm-mac.cfg │ ├── experiment02-hourglass-gpu.cfg │ ├── experiment02-hourglass-mac.cfg │ ├── experiment02-sp-mobilenetv2-gpu.cfg │ ├── experiment02-sp-mobilenetv2-mac.cfg │ ├── experiment02-sp-resnet18-gpu.cfg │ ├── experiment02-sp-resnet18-mac.cfg │ ├── experiment04-cpm-sg1-gpu.cfg │ ├── experiment04-cpm-sg1-mac.cfg │ ├── experiment04-cpm-sg2-colab.cfg │ ├── experiment04-cpm-sg2-gpu.cfg │ ├── experiment04-cpm-sg2-mac.cfg │ ├── experiment04-cpm-sg3-colab.cfg │ ├── experiment04-cpm-sg3-gpu.cfg │ ├── experiment04-cpm-sg4-colab.cfg │ ├── experiment04-cpm-sg4-gpu.cfg │ ├── experiment04-cpm-sg5-colab.cfg │ ├── experiment04-cpm-sg5-gpu.cfg │ ├── experiment05-sp-mv2-colab.cfg │ └── experiment05-sp-mv2-mac.cfg ├── convert_to_coreml.py ├── convert_to_tflite.py ├── data_loader ├── data_loader.py ├── dataset_augment.py ├── dataset_prepare.py └── pose_image_processor.py ├── data_processing ├── filter_aichallenge2018_only_single_person.py ├── filter_coco_only_single_person.py └── merge_two_datasets.py ├── downloader.py ├── evaluate.py ├── evaluate_tflite.py ├── model_provider.py ├── models ├── common.py ├── mobilenet.py ├── mobilenetv2.py ├── mobilenetv3.py ├── mv2_cpm.py ├── mv2_hourglass.py ├── resnet.py ├── resneta.py ├── resnetd.py ├── senet.py ├── simplepose_coco.py └── simpleposemobile_coco.py ├── requirements.txt ├── save_result_as_image.py ├── train.py └── train_all.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/README.md -------------------------------------------------------------------------------- /callbacks_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/callbacks_model.py -------------------------------------------------------------------------------- /common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/common.py -------------------------------------------------------------------------------- /config/dataset/ai_challenger-colab.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/dataset/ai_challenger-colab.cfg -------------------------------------------------------------------------------- /config/dataset/ai_challenger-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/dataset/ai_challenger-gpu.cfg -------------------------------------------------------------------------------- /config/dataset/ai_challenger-mac.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/dataset/ai_challenger-mac.cfg -------------------------------------------------------------------------------- /config/dataset/coco2017-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/dataset/coco2017-gpu.cfg -------------------------------------------------------------------------------- /config/dataset/coco_single_person_only-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/dataset/coco_single_person_only-gpu.cfg -------------------------------------------------------------------------------- /config/dataset/receipt5982-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/dataset/receipt5982-gpu.cfg -------------------------------------------------------------------------------- /config/training/coco_single_experiment01-cpm-sg4-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/coco_single_experiment01-cpm-sg4-gpu.cfg -------------------------------------------------------------------------------- /config/training/experiment01-colab.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment01-colab.cfg -------------------------------------------------------------------------------- /config/training/experiment01-mac.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment01-mac.cfg -------------------------------------------------------------------------------- /config/training/experiment01.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment01.cfg -------------------------------------------------------------------------------- /config/training/experiment02-cpm-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment02-cpm-gpu.cfg -------------------------------------------------------------------------------- /config/training/experiment02-cpm-mac.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment02-cpm-mac.cfg -------------------------------------------------------------------------------- /config/training/experiment02-hourglass-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment02-hourglass-gpu.cfg -------------------------------------------------------------------------------- /config/training/experiment02-hourglass-mac.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment02-hourglass-mac.cfg -------------------------------------------------------------------------------- /config/training/experiment02-sp-mobilenetv2-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment02-sp-mobilenetv2-gpu.cfg -------------------------------------------------------------------------------- /config/training/experiment02-sp-mobilenetv2-mac.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment02-sp-mobilenetv2-mac.cfg -------------------------------------------------------------------------------- /config/training/experiment02-sp-resnet18-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment02-sp-resnet18-gpu.cfg -------------------------------------------------------------------------------- /config/training/experiment02-sp-resnet18-mac.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment02-sp-resnet18-mac.cfg -------------------------------------------------------------------------------- /config/training/experiment04-cpm-sg1-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment04-cpm-sg1-gpu.cfg -------------------------------------------------------------------------------- /config/training/experiment04-cpm-sg1-mac.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment04-cpm-sg1-mac.cfg -------------------------------------------------------------------------------- /config/training/experiment04-cpm-sg2-colab.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment04-cpm-sg2-colab.cfg -------------------------------------------------------------------------------- /config/training/experiment04-cpm-sg2-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment04-cpm-sg2-gpu.cfg -------------------------------------------------------------------------------- /config/training/experiment04-cpm-sg2-mac.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment04-cpm-sg2-mac.cfg -------------------------------------------------------------------------------- /config/training/experiment04-cpm-sg3-colab.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment04-cpm-sg3-colab.cfg -------------------------------------------------------------------------------- /config/training/experiment04-cpm-sg3-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment04-cpm-sg3-gpu.cfg -------------------------------------------------------------------------------- /config/training/experiment04-cpm-sg4-colab.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment04-cpm-sg4-colab.cfg -------------------------------------------------------------------------------- /config/training/experiment04-cpm-sg4-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment04-cpm-sg4-gpu.cfg -------------------------------------------------------------------------------- /config/training/experiment04-cpm-sg5-colab.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment04-cpm-sg5-colab.cfg -------------------------------------------------------------------------------- /config/training/experiment04-cpm-sg5-gpu.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment04-cpm-sg5-gpu.cfg -------------------------------------------------------------------------------- /config/training/experiment05-sp-mv2-colab.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment05-sp-mv2-colab.cfg -------------------------------------------------------------------------------- /config/training/experiment05-sp-mv2-mac.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/config/training/experiment05-sp-mv2-mac.cfg -------------------------------------------------------------------------------- /convert_to_coreml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/convert_to_coreml.py -------------------------------------------------------------------------------- /convert_to_tflite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/convert_to_tflite.py -------------------------------------------------------------------------------- /data_loader/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/data_loader/data_loader.py -------------------------------------------------------------------------------- /data_loader/dataset_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/data_loader/dataset_augment.py -------------------------------------------------------------------------------- /data_loader/dataset_prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/data_loader/dataset_prepare.py -------------------------------------------------------------------------------- /data_loader/pose_image_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/data_loader/pose_image_processor.py -------------------------------------------------------------------------------- /data_processing/filter_aichallenge2018_only_single_person.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/data_processing/filter_aichallenge2018_only_single_person.py -------------------------------------------------------------------------------- /data_processing/filter_coco_only_single_person.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/data_processing/filter_coco_only_single_person.py -------------------------------------------------------------------------------- /data_processing/merge_two_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/data_processing/merge_two_datasets.py -------------------------------------------------------------------------------- /downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/downloader.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/evaluate.py -------------------------------------------------------------------------------- /evaluate_tflite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/evaluate_tflite.py -------------------------------------------------------------------------------- /model_provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/model_provider.py -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/common.py -------------------------------------------------------------------------------- /models/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/mobilenet.py -------------------------------------------------------------------------------- /models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/mobilenetv2.py -------------------------------------------------------------------------------- /models/mobilenetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/mobilenetv3.py -------------------------------------------------------------------------------- /models/mv2_cpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/mv2_cpm.py -------------------------------------------------------------------------------- /models/mv2_hourglass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/mv2_hourglass.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resneta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/resneta.py -------------------------------------------------------------------------------- /models/resnetd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/resnetd.py -------------------------------------------------------------------------------- /models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/senet.py -------------------------------------------------------------------------------- /models/simplepose_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/simplepose_coco.py -------------------------------------------------------------------------------- /models/simpleposemobile_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/models/simpleposemobile_coco.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/requirements.txt -------------------------------------------------------------------------------- /save_result_as_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/save_result_as_image.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/train.py -------------------------------------------------------------------------------- /train_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tucan9389/tf2-mobile-2d-single-pose-estimation/HEAD/train_all.sh --------------------------------------------------------------------------------