Spaces:
Runtime error
Runtime error
fifa-tryon-demo
/
Self-Correction-Human-Parsing-for-ACGPN
/mhp_extension
/coco_style_annotation_creator
/test_human2coco_format.py
| import argparse | |
| import datetime | |
| import json | |
| import os | |
| from PIL import Image | |
| import pycococreatortools | |
| def get_arguments(): | |
| parser = argparse.ArgumentParser(description="transform mask annotation to coco annotation") | |
| parser.add_argument("--dataset", type=str, default='CIHP', help="name of dataset (CIHP, MHPv2 or VIP)") | |
| parser.add_argument("--json_save_dir", type=str, default='../data/CIHP/annotations', | |
| help="path to save coco-style annotation json file") | |
| parser.add_argument("--test_img_dir", type=str, default='../data/CIHP/Testing/Images', | |
| help="test image path") | |
| return parser.parse_args() | |
| args = get_arguments() | |
| INFO = { | |
| "description": args.dataset + "Dataset", | |
| "url": "", | |
| "version": "", | |
| "year": 2020, | |
| "contributor": "yunqiuxu", | |
| "date_created": datetime.datetime.utcnow().isoformat(' ') | |
| } | |
| LICENSES = [ | |
| { | |
| "id": 1, | |
| "name": "", | |
| "url": "" | |
| } | |
| ] | |
| CATEGORIES = [ | |
| { | |
| 'id': 1, | |
| 'name': 'person', | |
| 'supercategory': 'person', | |
| }, | |
| ] | |
| def main(args): | |
| coco_output = { | |
| "info": INFO, | |
| "licenses": LICENSES, | |
| "categories": CATEGORIES, | |
| "images": [], | |
| "annotations": [] | |
| } | |
| image_id = 1 | |
| for image_name in os.listdir(args.test_img_dir): | |
| image = Image.open(os.path.join(args.test_img_dir, image_name)) | |
| image_info = pycococreatortools.create_image_info( | |
| image_id, image_name, image.size | |
| ) | |
| coco_output["images"].append(image_info) | |
| image_id += 1 | |
| if not os.path.exists(os.path.join(args.json_save_dir)): | |
| os.mkdir(os.path.join(args.json_save_dir)) | |
| with open('{}/{}.json'.format(args.json_save_dir, args.dataset), 'w') as output_json_file: | |
| json.dump(coco_output, output_json_file) | |
| if __name__ == "__main__": | |
| main(args) | |