Fix class id in yaml file.

Author: darouwan

general_json2yolo.py

@@ -257,14 +257,14 @@ def convert_coco_json(json_dir='../coco/annotations/', use_segments=False, cls91
     categories_dict = {}
 
     # Import json
-    for json_file in sorted(Path(json_dir).resolve().glob('*.json')):
+    for json_file in sorted(Path(json_dir).resolve().glob('*coco*.json')):
         fn = Path(save_dir) / 'labels' / json_file.stem.replace('instances_', '')  # folder name
         fn.mkdir()
         with open(json_file) as f:
             data = json.load(f)
 
         # Create categories_dict
-        categories_dict = {item['id']: item['name'] for item in data['categories']}
+        categories_dict = {item['id'] - 1: item['name'] for item in data['categories']}
 
         # Create image dict
         images = {'%g' % x['id']: x for x in data['images']}
@@ -399,7 +399,7 @@ def delete_dsstore(path='../datasets'):
     source = 'COCO'
 
     if source == 'COCO':
-        convert_coco_json('../datasets/coco/annotations',  # directory with *.json
+        convert_coco_json(r'C:\Users\1000276203\OneDrive - Western Digital\SDSS_DPAOI_CV\data\pure_dataset_memory\new chipping',  # directory with *.json
                           use_segments=True,
                           cls91to80=True)