diff --git a/src/super_gradients/training/utils/detection_utils.py b/src/super_gradients/training/utils/detection_utils.py index 5a267ed1a2..339ba643f1 100755 --- a/src/super_gradients/training/utils/detection_utils.py +++ b/src/super_gradients/training/utils/detection_utils.py @@ -538,7 +538,7 @@ def visualize_batch( 0 for invisible, 1 for fully opaque """ image_np = undo_preprocessing_func(image_tensor.detach()) - targets = DetectionVisualization._scaled_ccwh_to_xyxy(target_boxes.detach().cpu().numpy(), *image_np.shape[1:3], image_scale) + targets = DetectionVisualization._scaled_ccwh_to_xyxy(target_boxes.detach().cpu().numpy().copy(), *image_np.shape[1:3], image_scale) if pred_boxes is None: pred_boxes = [None for _ in range(image_np.shape[0])]