Skip to content

Commit

Permalink
Fixed wandb logger KeyError (#6637)
Browse files Browse the repository at this point in the history
  • Loading branch information
imyhxy authored Feb 14, 2022
1 parent 5389300 commit ca0a007
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion utils/loggers/wandb/wandb_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit ca0a007

Please sign in to comment.