diff --git a/torchgeo/datasets/vhr10.py b/torchgeo/datasets/vhr10.py index 5628bcd16d6..67f9407ddb9 100644 --- a/torchgeo/datasets/vhr10.py +++ b/torchgeo/datasets/vhr10.py @@ -400,7 +400,6 @@ def plot( if show_feats != 'boxes': skimage = lazy_import('skimage') - image = sample['image'].permute(1, 2, 0).numpy() boxes = sample['bbox_xyxy'].cpu().numpy() labels = sample['label'].cpu().numpy() if 'mask' in sample: