Accept issue #5

Author: SWHL

README.md

@@ -93,7 +93,15 @@
       ├── train.txt
       └── val.txt
       ```
-
+  - **train.txt**和**val.txt**中图像路径，以下两种均可：
+    - 相对于**根目录**的路径
+      ```text
+      dataset/YOLOV5/images/images(3).jpg
+      ```
+    - 相对于**dataset/YOLOV5**的相对路径
+      ```text
+      images/images(3).jpg
+      ```
   - 转换
       ```shell
     python yolov5_2_coco.py --dir_path dataset/YOLOV5 --mode_list train,val
@@ -102,17 +110,17 @@
     - `--mode_list`：指定生成的json，前提是要有对应的txt文件，可单独指定。（e.g. `train,val,test`）
 
   - 转换后目录结构（详情参见`dataset/YOLOV5_COCO_format`）：
-  ```text
-  YOLOV5_COCO_format
-  ├── annotations
-  │   ├── instances_train2017.json
-  │   └── instances_val2017.json
-  ├── train2017
-  │   ├── 000000000001.jpg
-  │   └── 000000000002.jpg  # 这个是背景图像
-  └── val2017
-      └── 000000000001.jpg
-  ```
+    ```text
+    YOLOV5_COCO_format
+    ├── annotations
+    │   ├── instances_train2017.json
+    │   └── instances_val2017.json
+    ├── train2017
+    │   ├── 000000000001.jpg
+    │   └── 000000000002.jpg  # 这个是背景图像
+    └── val2017
+        └── 000000000001.jpg
+    ```
 </details>
 
 #### YOLOV5 YAML描述文件 → COCO

docs/README_en.md

@@ -93,13 +93,21 @@ English | [简体中文](../README.md)
     ├── train.txt
     └── val.txt
     ```
-
+- The image paths in train.txt and val.txt can be either:
+  - Path relative to **root directory**:
+      ```text
+      dataset/YOLOV5/images/images(3).jpg
+      ```
+  - Path relative to **dataset/YOLOV5**:
+      ```text
+      images/images(3).jpg
+      ```
 - Convert
     ```shell
   python yolov5_2_coco.py --dir_path dataset/YOLOV5 --mode_list train,val
   ```
   - `--dir_path`: the directory where the collated dataset is located
-  - `--mode_list`: specify the generated json, provided that there is a corresponding txt file, which can be specified separately. (e.g. `-train,val,test`)
+  - `--mode_list`: specify the generated json, provided that there is a corresponding txt file, which can be specified separately. (e.g. `train,val,test`)
 
 - The structure of the converted directory (see `dataset/YOLOV5_COCO_format` for details)
     ```text

yolov5_2_coco.py

@@ -117,6 +117,11 @@ def convert(self, img_list, save_img_dir, mode):
 
     def get_image_info(self, img_path, img_id, save_img_dir):
         img_path = Path(img_path)
+        if not img_path.as_posix().__contains__(self.raw_data_dir.as_posix()):
+            # relative path (relative to the raw_data_dir)
+            # e.g. images/images(3).jpg
+            img_path = self.raw_data_dir / img_path
+
         self.verify_exists(img_path)
 
         new_img_name = f'{img_id:012d}.jpg'