fix bugs;add README

Author: goderent

README.md

@@ -48,6 +48,22 @@
         └── 000000000001.jpg
     ```
 
+#### YOLOV5 yaml数据描述文件转→COCO
+
+- YOLOV5 yaml 数据文件需要包含：
+    ```text
+    YOLOV5 yaml
+    ├── path(str, the root path)
+    ├── train(Train sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..])
+    ├── val(val sets, similar as train)
+
+    ```
+
+- 转换
+  ```shell
+  python yolov5_cfgfile_2_coco.py --cfg_file dataset/sample.yaml
+  ```
+
 #### 可视化COCO格式标注格式
   ```shell
   python coco_visual.py --vis_num 1 \

dataset/sample.yaml

@@ -0,0 +1,8 @@
+# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
+path: /the/root/path/to/data  # dataset root dir
+train: [data1/images/train/, data2/images/train/]
+val: [data1/images/val/, data2/images/val/]
+
+# Classes
+nc: 3  # number of classes
+names: ['class1', 'class2', 'class3']  # class names

yolov5_cfgfile_2_coco.py

@@ -65,6 +65,7 @@ def __init__(self, cfg_file):
         # 构建json内容结构
         self.type = 'instances'
         self.categories = []
+        self._get_category()
         self.annotation_id = 1
 
         self.info = {
@@ -87,11 +88,11 @@ def _get_category(self):
                 'id': i,
                 'name': category,
             })
-        self.categories.append({
-                'supercategory': 'Background',
-                'id': 0,
-                'name': 'Background',
-            })
+        # self.categories.append({
+        #         'supercategory': 'Background',
+        #         'id': 0,
+        #         'name': 'Background',
+        #     })
 
     def generate(self):
         self.train_files = self.getfiles(self.train_path)
@@ -101,9 +102,9 @@ def generate(self):
         self.gen_dataset(self.train_files, train_dest_dir,
                          self.coco_train_json, mode='train')
 
-        # val_dest_dir = Path(self.dst) / self.coco_val
-        # self.gen_dataset(self.valid_files, val_dest_dir,
-        #                  self.coco_val_json, mode='val')
+        val_dest_dir = Path(self.dst) / self.coco_val
+        self.gen_dataset(self.valid_files, val_dest_dir,
+                         self.coco_val_json, mode='val')
 
         print(f"The output directory is: {str(self.dst)}")