Skip to content

Commit 442689f

Browse files
committed
remove assert for image logger
1 parent 77e089b commit 442689f

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

yolo/utils/logging_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,7 @@ def on_validation_batch_end(self, trainer: Trainer, pl_module, outputs, batch, b
264264
image_hwc = einops.rearrange(image_chw, 'c h w -> h w c')
265265
image_hwc = kwimage.ensure_uint255(image_hwc)
266266

267-
assert bx == 0, 'not handling multiple per batch'
267+
# assert bx == 0, 'not handling multiple per batch'
268268
true_dets = tensor_to_kwimage(gt_boxes).numpy()
269269
pred_dets = tensor_to_kwimage(pred_boxes).numpy()
270270
pred_dets = pred_dets.non_max_supress(thresh=0.3)

0 commit comments

Comments
 (0)