fix weight s3 uri for windows (#63)

* fix weight s3 uri for windows * fix a wandb error
fcakyon · Nov 29, 2021 · 6a8a50e · 6a8a50e
1 parent d3c306e
commit 6a8a50e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/yolov5/utils/loggers/__init__.py b/yolov5/utils/loggers/__init__.py
@@ -64,7 +64,7 @@ def __init__(self, save_dir=None, weights=None, opt=None, hyp=None, logger=None,
             self.class_name_keys = ['metrics/' + name + '_mAP_50' for name in class_names]
         else:
             self.class_name_keys = ['val/' + name + '_mAP_50' for name in class_names]
-        self.s3_weight_folder = None if not opt.s3_upload_dir else "s3://" + str(Path(opt.s3_upload_dir.replace("s3://","")) / save_dir.name / "weights")
+        self.s3_weight_folder = None if not opt.s3_upload_dir else "s3://" + str(Path(opt.s3_upload_dir.replace("s3://","")) / save_dir.name / "weights").replace(os.sep, '/')
 
         # Message
         if not wandb:

diff --git a/yolov5/utils/loggers/wandb/wandb_utils.py b/yolov5/utils/loggers/wandb/wandb_utils.py
@@ -487,7 +487,7 @@ def end_epoch(self, best_result=False):
         if self.wandb_run:
             with all_logging_disabled():
                 if self.bbox_media_panel_images:
-                    self.log_dict["Bounding Box Debugger/Images"] = self.bbox_media_panel_images
+                    self.log_dict["BoundingBoxDebugger"] = self.bbox_media_panel_images
                 wandb.log(self.log_dict)
                 self.log_dict = {}
                 self.bbox_media_panel_images = []