├── .gitignore ├── LICENSE ├── README.md ├── demo ├── blouse.png ├── dress.png ├── outwear.png ├── skirt.png └── trousers.png ├── experiements.xlsx └── src ├── __init__.py ├── capb_parser.py ├── check_duplicated_image.py ├── config.py ├── kpda_parser.py ├── kpdetector ├── __init__.py ├── concatenate_results.py ├── predict.py ├── predict_ensemble.py └── verify_result.py ├── lr_scheduler.py ├── offline_evaluation.py ├── pytorch_utils.py ├── stage1 ├── __init__.py ├── data_generator.py ├── focal_loss.py ├── fpn.py ├── label_encoder.py ├── predict.py ├── retinanet.py └── trainval.py ├── stage2 ├── __init__.py ├── autorun.sh ├── cascade_pyramid_network.py ├── cascade_pyramid_network_v10.py ├── cascade_pyramid_network_v11.py ├── cascade_pyramid_network_v12.py ├── cascade_pyramid_network_v14.py ├── cascade_pyramid_network_v2.py ├── cascade_pyramid_network_v5.py ├── cascade_pyramid_network_v6.py ├── cascade_pyramid_network_v7.py ├── cascade_pyramid_network_v8.py ├── data_generator.py ├── keypoint_encoder.py ├── predict_ensemble.py ├── predict_one.py ├── trainval.py └── viserrloss.py ├── stage2v13 ├── __init__.py ├── cascade_pyramid_network_v13.py ├── data_generator.py ├── trainval.py └── viserrloss_v13.py ├── stage2v15 ├── __init__.py ├── cascade_pyramid_network_v15.py └── nasnet.py ├── stage2v2 ├── __init__.py ├── autorun.sh ├── cascade_pyramid_network_v2.py ├── data_generator.py ├── trainval.py └── viserrloss_v2.py ├── stage2v3 ├── __init__.py ├── autorun.sh ├── cascade_pyramid_network_v3.py ├── trainval.py └── viserrloss_v3.py ├── stage2v4 ├── __init__.py ├── cascade_pyramid_network_v4.py └── inceptionresnetv2.py ├── stage2v9 ├── __init__.py ├── cascade_pyramid_network_v9.py └── senet.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea/ 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/README.md -------------------------------------------------------------------------------- /demo/blouse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/demo/blouse.png -------------------------------------------------------------------------------- /demo/dress.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/demo/dress.png -------------------------------------------------------------------------------- /demo/outwear.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/demo/outwear.png -------------------------------------------------------------------------------- /demo/skirt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/demo/skirt.png -------------------------------------------------------------------------------- /demo/trousers.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/demo/trousers.png -------------------------------------------------------------------------------- /experiements.xlsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/experiements.xlsx -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/capb_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/capb_parser.py -------------------------------------------------------------------------------- /src/check_duplicated_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/check_duplicated_image.py -------------------------------------------------------------------------------- /src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/config.py -------------------------------------------------------------------------------- /src/kpda_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/kpda_parser.py -------------------------------------------------------------------------------- /src/kpdetector/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/kpdetector/concatenate_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/kpdetector/concatenate_results.py -------------------------------------------------------------------------------- /src/kpdetector/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/kpdetector/predict.py -------------------------------------------------------------------------------- /src/kpdetector/predict_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/kpdetector/predict_ensemble.py -------------------------------------------------------------------------------- /src/kpdetector/verify_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/kpdetector/verify_result.py -------------------------------------------------------------------------------- /src/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/lr_scheduler.py -------------------------------------------------------------------------------- /src/offline_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/offline_evaluation.py -------------------------------------------------------------------------------- /src/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/pytorch_utils.py -------------------------------------------------------------------------------- /src/stage1/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/stage1/data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage1/data_generator.py -------------------------------------------------------------------------------- /src/stage1/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage1/focal_loss.py -------------------------------------------------------------------------------- /src/stage1/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage1/fpn.py -------------------------------------------------------------------------------- /src/stage1/label_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage1/label_encoder.py -------------------------------------------------------------------------------- /src/stage1/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage1/predict.py -------------------------------------------------------------------------------- /src/stage1/retinanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage1/retinanet.py -------------------------------------------------------------------------------- /src/stage1/trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage1/trainval.py -------------------------------------------------------------------------------- /src/stage2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/stage2/autorun.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/autorun.sh -------------------------------------------------------------------------------- /src/stage2/cascade_pyramid_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/cascade_pyramid_network.py -------------------------------------------------------------------------------- /src/stage2/cascade_pyramid_network_v10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/cascade_pyramid_network_v10.py -------------------------------------------------------------------------------- /src/stage2/cascade_pyramid_network_v11.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/cascade_pyramid_network_v11.py -------------------------------------------------------------------------------- /src/stage2/cascade_pyramid_network_v12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/cascade_pyramid_network_v12.py -------------------------------------------------------------------------------- /src/stage2/cascade_pyramid_network_v14.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/cascade_pyramid_network_v14.py -------------------------------------------------------------------------------- /src/stage2/cascade_pyramid_network_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/cascade_pyramid_network_v2.py -------------------------------------------------------------------------------- /src/stage2/cascade_pyramid_network_v5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/cascade_pyramid_network_v5.py -------------------------------------------------------------------------------- /src/stage2/cascade_pyramid_network_v6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/cascade_pyramid_network_v6.py -------------------------------------------------------------------------------- /src/stage2/cascade_pyramid_network_v7.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/cascade_pyramid_network_v7.py -------------------------------------------------------------------------------- /src/stage2/cascade_pyramid_network_v8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/cascade_pyramid_network_v8.py -------------------------------------------------------------------------------- /src/stage2/data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/data_generator.py -------------------------------------------------------------------------------- /src/stage2/keypoint_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/keypoint_encoder.py -------------------------------------------------------------------------------- /src/stage2/predict_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/predict_ensemble.py -------------------------------------------------------------------------------- /src/stage2/predict_one.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/predict_one.py -------------------------------------------------------------------------------- /src/stage2/trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/trainval.py -------------------------------------------------------------------------------- /src/stage2/viserrloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2/viserrloss.py -------------------------------------------------------------------------------- /src/stage2v13/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v13/__init__.py -------------------------------------------------------------------------------- /src/stage2v13/cascade_pyramid_network_v13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v13/cascade_pyramid_network_v13.py -------------------------------------------------------------------------------- /src/stage2v13/data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v13/data_generator.py -------------------------------------------------------------------------------- /src/stage2v13/trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v13/trainval.py -------------------------------------------------------------------------------- /src/stage2v13/viserrloss_v13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v13/viserrloss_v13.py -------------------------------------------------------------------------------- /src/stage2v15/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/stage2v15/cascade_pyramid_network_v15.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v15/cascade_pyramid_network_v15.py -------------------------------------------------------------------------------- /src/stage2v15/nasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v15/nasnet.py -------------------------------------------------------------------------------- /src/stage2v2/__init__.py: -------------------------------------------------------------------------------- 1 | # compute loss for each level -------------------------------------------------------------------------------- /src/stage2v2/autorun.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v2/autorun.sh -------------------------------------------------------------------------------- /src/stage2v2/cascade_pyramid_network_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v2/cascade_pyramid_network_v2.py -------------------------------------------------------------------------------- /src/stage2v2/data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v2/data_generator.py -------------------------------------------------------------------------------- /src/stage2v2/trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v2/trainval.py -------------------------------------------------------------------------------- /src/stage2v2/viserrloss_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v2/viserrloss_v2.py -------------------------------------------------------------------------------- /src/stage2v3/__init__.py: -------------------------------------------------------------------------------- 1 | # Second refine net after the first one -------------------------------------------------------------------------------- /src/stage2v3/autorun.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v3/autorun.sh -------------------------------------------------------------------------------- /src/stage2v3/cascade_pyramid_network_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v3/cascade_pyramid_network_v3.py -------------------------------------------------------------------------------- /src/stage2v3/trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v3/trainval.py -------------------------------------------------------------------------------- /src/stage2v3/viserrloss_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v3/viserrloss_v3.py -------------------------------------------------------------------------------- /src/stage2v4/__init__.py: -------------------------------------------------------------------------------- 1 | # use Inception-Resnet V2 as backbone network -------------------------------------------------------------------------------- /src/stage2v4/cascade_pyramid_network_v4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v4/cascade_pyramid_network_v4.py -------------------------------------------------------------------------------- /src/stage2v4/inceptionresnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v4/inceptionresnetv2.py -------------------------------------------------------------------------------- /src/stage2v9/__init__.py: -------------------------------------------------------------------------------- 1 | # use SENet154 as backbone network -------------------------------------------------------------------------------- /src/stage2v9/cascade_pyramid_network_v9.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v9/cascade_pyramid_network_v9.py -------------------------------------------------------------------------------- /src/stage2v9/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/stage2v9/senet.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gathierry/FashionAI-KeyPointsDetectionOfApparel/HEAD/src/utils.py --------------------------------------------------------------------------------