From c6794763474a90c5a4e02365974866ea59078b81 Mon Sep 17 00:00:00 2001 From: imyhxy Date: Tue, 15 Feb 2022 00:08:36 +0800 Subject: [PATCH] Fixed wandb logger KeyError (#6637) --- utils/loggers/wandb/wandb_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/loggers/wandb/wandb_utils.py b/utils/loggers/wandb/wandb_utils.py index e97286876bbf..3835436543d2 100644 --- a/utils/loggers/wandb/wandb_utils.py +++ b/utils/loggers/wandb/wandb_utils.py @@ -487,7 +487,7 @@ def val_one_image(self, pred, predn, path, names, im): if self.current_epoch % self.bbox_interval == 0: box_data = [{"position": {"minX": xyxy[0], "minY": xyxy[1], "maxX": xyxy[2], "maxY": xyxy[3]}, "class_id": int(cls), - "box_caption": f"{names[cls]} {conf:.3f}", + "box_caption": f"{names[int(cls)]} {conf:.3f}", "scores": {"class_score": conf}, "domain": "pixel"} for *xyxy, conf, cls in pred.tolist()] boxes = {"predictions": {"box_data": box_data, "class_labels": names}} # inference-space