Skip to content

Commit

Permalink
Fix whitespace issue in VHR10 dataset handling
Browse files Browse the repository at this point in the history
  • Loading branch information
robmarkcole committed May 28, 2024
1 parent a0e17f0 commit 1baac80
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchgeo/datasets/vhr10.py
Original file line number Diff line number Diff line change
Expand Up @@ -246,7 +246,7 @@ def __getitem__(self, index: int) -> dict[str, Any]:
sample = self.coco_convert(sample)
sample['labels'] = sample['label']['labels']
sample['boxes'] = sample['label']['boxes']
sample['masks'] = sample['label']['masks']
sample['masks'] = sample['label']['masks']
del sample['label']

if self.transforms is not None:
Expand Down

0 comments on commit 1baac80

Please sign in to comment.