报错解决：ValueError: Media Bounding Box Debugger/Images is invalid.Please remove invalid

最新推荐文章于 2024-03-13 10:08:10 发布

Vv_esfs

最新推荐文章于 2024-03-13 10:08:10 发布

阅读量1.2k

点赞数 1

分类专栏： yolov5 文章标签：深度学习人工智能

本文链接：https://blog.csdn.net/Vv_esfs/article/details/122144496

版权

yolov5 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

在进行yolov5进行目标检测进行训练时，运行train.py，产生报错。错误内容：ValueError: Media Bounding Box Debugger/Images is invalid. Please remove invalid filename characters

报错原因：

解决方法

1、关闭wandb

在utils\loggers\wandb\wandb_utils.py

try:
    import wandb

    assert hasattr(wandb, '__version__')  # verify package import not local dir
    if pkg.parse_version(wandb.__version__) >= pkg.parse_version('0.12.2') and RANK in [0, -1]:
        wandb.login(timeout=30)
except (ImportError, AssertionError):
    wandb = None

后面增加 wandb = None

try:
    import wandb

    assert hasattr(wandb, '__version__')  # verify package import not local dir
    if pkg.parse_version(wandb.__version__) >= pkg.parse_version('0.12.2') and RANK in [0, -1]:
        wandb.login(timeout=30)
except (ImportError, AssertionError):
    wandb = None
wandb = None

2、重新安装wandb可能可以解决，并没有尝试

3、修改utils/loggers/wandb/wandb_utils.py

查找 def end_epoch 进行修改

    def end_epoch(self, best_result=False):
        """
        commit the log_dict, model artifacts and Tables to W&B and flush the log_dict.

        arguments:
        best_result (boolean): Boolean representing if the result of this evaluation is best or not
        """
        if self.wandb_run:
            with all_logging_disabled():
                if self.bbox_media_panel_images:
                    sself.log_dict["Bounding Box Debugger/Images"] = self.bbox_media_panel_images
                wandb.log(self.log_dict)
                self.log_dict = {}
                self.bbox_media_panel_images = []
            if self.result_artifact:
                self.result_artifact.add(self.result_table, 'result')
                wandb.log_artifact(self.result_artifact, aliases=['latest', 'last', 'epoch ' + str(self.current_epoch),
                                                                  ('best' if best_result else '')])

                wandb.log({"evaluation": self.result_table})
                self.result_table = wandb.Table(["epoch", "id", "ground truth", "prediction", "avg_confidence"])
                self.result_artifact = wandb.Artifact("run_" + wandb.run.id + "_progress", "evaluation")

修改为

    def end_epoch(self, best_result=False):
        """
        commit the log_dict, model artifacts and Tables to W&B and flush the log_dict.

        arguments:
        best_result (boolean): Boolean representing if the result of this evaluation is best or not
        """
        if self.wandb_run:
            with all_logging_disabled():
                if 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 = []
            if self.result_artifact:
                self.result_artifact.add(self.result_table, 'result')
                wandb.log_artifact(self.result_artifact, aliases=['latest', 'last', 'epoch ' + str(self.current_epoch),
                                                                  ('best' if best_result else '')])

                wandb.log({"evaluation": self.result_table})
                self.result_table = wandb.Table(["epoch", "id", "ground truth", "prediction", "avg_confidence"])
                self.result_artifact = wandb.Artifact("run_" + wandb.run.id + "_progress", "evaluation")