1. Generate classes.yaml

2. Remove coco suffix in output folder.

Author: darouwan

general_json2yolo.py

@@ -1,5 +1,7 @@
 import contextlib
 import json
+import shutil
+
 import yaml
 import cv2
 import pandas as pd
@@ -258,7 +260,7 @@ def convert_coco_json(json_dir='../coco/annotations/', use_segments=False, cls91
 
     # Import 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 = Path(save_dir) / 'labels' / json_file.stem.replace('instances_', '').replace('_coco','')  # folder name
         fn.mkdir()
         with open(json_file) as f:
             data = json.load(f)
@@ -322,6 +324,7 @@ def convert_coco_json(json_dir='../coco/annotations/', use_segments=False, cls91
 
         with open((Path(save_dir) / json_file.stem).with_suffix('.yaml'), "w") as f:
             yaml.dump(yaml_dict, f)
+        shutil.copy((Path(save_dir) / json_file.stem).with_suffix('.yaml'), (Path(save_dir) / 'classes').with_suffix('.yaml'))
 
 
 def min_index(arr1, arr2):
@@ -399,7 +402,7 @@ def delete_dsstore(path='../datasets'):
     source = 'COCO'
 
     if source == 'COCO':
-        convert_coco_json(r'C:\Users\1000276203\OneDrive - Western Digital\SDSS_DPAOI_CV\data\pure_dataset_memory\new chipping',  # directory with *.json
+        convert_coco_json(r'C:\Users\1000276203\OneDrive - Western Digital\SDSS_DPAOI_CV\data\new chipping',  # directory with *.json
                           use_segments=True,
                           cls91to80=True)