fix(demo & tools): path issues of Windows system (#1100)

fix(demo & tools): path issues of Windows system
Megvii-BaseDetection · Jan 27, 2022 · 6b49fd2 · 6b49fd2
1 parent 856bf0f
commit 6b49fd2
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/demo/MegEngine/python/demo.py b/demo/MegEngine/python/demo.py
@@ -182,7 +182,7 @@ def imageflow_demo(predictor, vis_folder, current_time, args):
     )
     os.makedirs(save_folder, exist_ok=True)
     if args.demo == "video":
-        save_path = os.path.join(save_folder, args.path.split("/")[-1])
+        save_path = os.path.join(save_folder, os.path.basename(args.path))
     else:
         save_path = os.path.join(save_folder, "camera.mp4")
     logger.info(f"video save_path is {save_path}")

diff --git a/demo/ONNXRuntime/onnx_inference.py b/demo/ONNXRuntime/onnx_inference.py
@@ -88,5 +88,5 @@ def make_parser():
                          conf=args.score_thr, class_names=COCO_CLASSES)
 
     mkdir(args.output_dir)
-    output_path = os.path.join(args.output_dir, args.image_path.split("/")[-1])
+    output_path = os.path.join(args.output_dir, os.path.basename(args.image_path))
     cv2.imwrite(output_path, origin_img)
diff --git a/demo/OpenVINO/python/openvino_inference.py b/demo/OpenVINO/python/openvino_inference.py
@@ -148,7 +148,7 @@ def main():
                          conf=args.score_thr, class_names=COCO_CLASSES)
 
     mkdir(args.output_dir)
-    output_path = os.path.join(args.output_dir, args.input.split("/")[-1])
+    output_path = os.path.join(args.output_dir, os.path.basename(args.input))
     cv2.imwrite(output_path, origin_img)
 
 

diff --git a/tools/demo.py b/tools/demo.py
@@ -216,7 +216,7 @@ def imageflow_demo(predictor, vis_folder, current_time, args):
     )
     os.makedirs(save_folder, exist_ok=True)
     if args.demo == "video":
-        save_path = os.path.join(save_folder, args.path.split("/")[-1])
+        save_path = os.path.join(save_folder, os.path.basename(args.path))
     else:
         save_path = os.path.join(save_folder, "camera.mp4")
     logger.info(f"video save_path is {save_path}")

diff --git a/yolox/data/datasets/coco.py b/yolox/data/datasets/coco.py
@@ -93,7 +93,7 @@ def _cache_images(self):
         )
         max_h = self.img_size[0]
         max_w = self.img_size[1]
-        cache_file = self.data_dir + "/img_resized_cache_" + self.name + ".array"
+        cache_file = os.path.join(self.data_dir, f"img_resized_cache_{self.name}.array")
         if not os.path.exists(cache_file):
             logger.info(
                 "Caching images for the first time. This might take about 20 minutes for COCO"

diff --git a/yolox/data/datasets/voc.py b/yolox/data/datasets/voc.py
@@ -149,7 +149,7 @@ def _cache_images(self):
         )
         max_h = self.img_size[0]
         max_w = self.img_size[1]
-        cache_file = self.root + "/img_resized_cache_" + self.name + ".array"
+        cache_file = os.path.join(self.root, f"img_resized_cache_{self.name}.array")
         if not os.path.exists(cache_file):
             logger.info(
                 "Caching images for the first time. This might take about 3 minutes for VOC"