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

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}")

demo/ONNXRuntime/onnx_inference.py

@@ -88,5 +88,5 @@ if __name__ == '__main__':
                          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)

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)
 
 

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}")

yolox/data/datasets/coco.py

@@ -93,7 +93,7 @@ class COCODataset(Dataset):
         )
         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"

yolox/data/datasets/voc.py

@@ -149,7 +149,7 @@ class VOCDetection(Dataset):
         )
         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"