Skip to content

Commit a7929d0

Browse files
committed
remove assert for image logger
1 parent 2f37936 commit a7929d0

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
@@ -249,7 +249,7 @@ def on_validation_batch_end(self, trainer: Trainer, pl_module, outputs, batch, b
249249
image_hwc = einops.rearrange(image_chw, 'c h w -> h w c')
250250
image_hwc = kwimage.ensure_uint255(image_hwc)
251251

252-
assert bx == 0, 'not handling multiple per batch'
252+
# assert bx == 0, 'not handling multiple per batch'
253253
true_dets = tensor_to_kwimage(gt_boxes).numpy()
254254
pred_dets = tensor_to_kwimage(pred_boxes).numpy()
255255
pred_dets = pred_dets.non_max_supress(thresh=0.3)

0 commit comments

Comments
 (0)