From 55376ec3b8900962397ed6807f40515b762ccfbc Mon Sep 17 00:00:00 2001 From: katsunori waragai Date: Mon, 3 Jul 2023 15:29:01 +0900 Subject: [PATCH 1/4] use nested folder --- README.md | 4 ++++ yolox/data/datasets/coco.py | 7 +++++++ 2 files changed, 11 insertions(+) diff --git a/README.md b/README.md index 18d9d4040..41a47055f 100644 --- a/README.md +++ b/README.md @@ -253,3 +253,7 @@ It is hoped that every AI practitioner in the world will stick to the concept of 没有孙剑博士的指导，YOLOX也不会问世并开源给社区使用。 孙剑博士的离去是CV领域的一大损失，我们在此特别添加了这个部分来表达对我们的“船长”孙老师的纪念和哀思。 希望世界上的每个AI从业者秉持着“持续创新拓展认知边界，非凡科技成就产品价值”的观念，一路向前。 + +## local custimze by GX +- 階層的なフォルダ構成での画像ファイルの読み込みを可能にするための改変。 +- 従来のflatなフォルダ構成でファイルがある場合には、階層的なフォルダ構成のための拡張は動作させない。 diff --git a/yolox/data/datasets/coco.py b/yolox/data/datasets/coco.py index 8d19047a2..59e51628a 100644 --- a/yolox/data/datasets/coco.py +++ b/yolox/data/datasets/coco.py @@ -3,6 +3,7 @@ # Copyright (c) Megvii, Inc. and its affiliates. import copy import os +from pathlib import Path import cv2 import numpy as np @@ -143,8 +144,14 @@ def load_resized_img(self, index): def load_image(self, index): file_name = self.annotations[index][3] + # 従来どおりのflatなフォルダ構成 img_file = os.path.join(self.data_dir, self.name, file_name) + # 階層的なフォルダ構成の場合 + if not Path(img_file).is_file(): + img_file = os.path.join(self.data_dir, file_name) + assert Path(img_file).is_file() + img = cv2.imread(img_file) assert img is not None, f"file named {img_file} not found" From a83d08ea5cb102761661b6331af1a84748682c95 Mon Sep 17 00:00:00 2001 From: katsunori waragai Date: Mon, 3 Jul 2023 15:51:09 +0900 Subject: [PATCH 2/4] use os.path.isfile() --- yolox/data/datasets/coco.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/yolox/data/datasets/coco.py b/yolox/data/datasets/coco.py index 59e51628a..4722f6fdf 100644 --- a/yolox/data/datasets/coco.py +++ b/yolox/data/datasets/coco.py @@ -3,7 +3,6 @@ # Copyright (c) Megvii, Inc. and its affiliates. import copy import os -from pathlib import Path import cv2 import numpy as np @@ -144,13 +143,11 @@ def load_resized_img(self, index): def load_image(self, index): file_name = self.annotations[index][3] - # 従来どおりのflatなフォルダ構成 img_file = os.path.join(self.data_dir, self.name, file_name) - # 階層的なフォルダ構成の場合 - if not Path(img_file).is_file(): + if not os.path.isfile(img_file): + # Hierarchical folder structure case img_file = os.path.join(self.data_dir, file_name) - assert Path(img_file).is_file() img = cv2.imread(img_file) assert img is not None, f"file named {img_file} not found" From 677805d91b535e02f455b57811cdda75d2a678c6 Mon Sep 17 00:00:00 2001 From: katsunori waragai Date: Mon, 3 Jul 2023 16:11:22 +0900 Subject: [PATCH 3/4] less Changes --- README.md | 3 --- 1 file changed, 3 deletions(-) diff --git a/README.md b/README.md index 41a47055f..8dbdaa050 100644 --- a/README.md +++ b/README.md @@ -254,6 +254,3 @@ It is hoped that every AI practitioner in the world will stick to the concept of 孙剑博士的离去是CV领域的一大损失，我们在此特别添加了这个部分来表达对我们的“船长”孙老师的纪念和哀思。 希望世界上的每个AI从业者秉持着“持续创新拓展认知边界，非凡科技成就产品价值”的观念，一路向前。 -## local custimze by GX -- 階層的なフォルダ構成での画像ファイルの読み込みを可能にするための改変。 -- 従来のflatなフォルダ構成でファイルがある場合には、階層的なフォルダ構成のための拡張は動作させない。 From 498b2bca6118f37fea70afa2fb1d641cf7b4640e Mon Sep 17 00:00:00 2001 From: katsunori waragai Date: Mon, 3 Jul 2023 16:11:56 +0900 Subject: [PATCH 4/4] less Changes --- README.md | 1 - 1 file changed, 1 deletion(-) diff --git a/README.md b/README.md index 8dbdaa050..18d9d4040 100644 --- a/README.md +++ b/README.md @@ -253,4 +253,3 @@ It is hoped that every AI practitioner in the world will stick to the concept of 没有孙剑博士的指导，YOLOX也不会问世并开源给社区使用。 孙剑博士的离去是CV领域的一大损失，我们在此特别添加了这个部分来表达对我们的“船长”孙老师的纪念和哀思。 希望世界上的每个AI从业者秉持着“持续创新拓展认知边界，非凡科技成就产品价值”的观念，一路向前。 -