|
|
@@ -156,7 +156,7 @@ def test(data, |
|
|
|
"domain": "pixel"} for *xyxy, conf, cls in pred.tolist()] |
|
|
|
boxes = {"predictions": {"box_data": box_data, "class_labels": names}} # inference-space |
|
|
|
wandb_images.append(wandb_logger.wandb.Image(img[si], boxes=boxes, caption=path.name)) |
|
|
|
wandb_logger.log_training_progress(predn, path, names) # logs dsviz tables |
|
|
|
wandb_logger.log_training_progress(predn, path, names) if wandb_logger and wandb_logger.wandb_run else None |
|
|
|
|
|
|
|
# Append to pycocotools JSON dictionary |
|
|
|
if save_json: |