Ayush Chaurasia
commited on
Commit
•
8f6e447
1
Parent(s):
a57f23d
Fix Indentation in test.py (#2614)
Browse files* Fix Indentation in test.py
* CI fix
* Comply with PEP8: 80 characters per line
test.py
CHANGED
@@ -156,7 +156,7 @@ def test(data,
|
|
156 |
"domain": "pixel"} for *xyxy, conf, cls in pred.tolist()]
|
157 |
boxes = {"predictions": {"box_data": box_data, "class_labels": names}} # inference-space
|
158 |
wandb_images.append(wandb_logger.wandb.Image(img[si], boxes=boxes, caption=path.name))
|
159 |
-
|
160 |
|
161 |
# Append to pycocotools JSON dictionary
|
162 |
if save_json:
|
|
|
156 |
"domain": "pixel"} for *xyxy, conf, cls in pred.tolist()]
|
157 |
boxes = {"predictions": {"box_data": box_data, "class_labels": names}} # inference-space
|
158 |
wandb_images.append(wandb_logger.wandb.Image(img[si], boxes=boxes, caption=path.name))
|
159 |
+
wandb_logger.log_training_progress(predn, path, names) if wandb_logger and wandb_logger.wandb_run else None
|
160 |
|
161 |
# Append to pycocotools JSON dictionary
|
162 |
if save_json:
|