Skip to content

Commit e2b5d86

Browse files
committed
Fix PADDataset name
1 parent db68141 commit e2b5d86

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

utils/PAD_datamodule.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
class PADDataModule(pl.LightningDataModule):
1818
# Documentation: https://pytorch-lightning.readthedocs.io/en/latest/extensions/datamodules.html
1919
'''
20-
PyTorch Lightning DataModule Wrapper for PatchesDataset
20+
PyTorch Lightning DataModule Wrapper for PADDataset
2121
'''
2222

2323
def __init__(
@@ -180,7 +180,7 @@ def setup(self, stage=None):
180180
coco_train = COCO(self.path_train)
181181
coco_val = COCO(self.path_val)
182182

183-
self.dataset_train = PatchesDataset(root_path_coco=self.root_path_coco,
183+
self.dataset_train = PADDataset(root_path_coco=self.root_path_coco,
184184
coco=coco_train,
185185
# transforms=transforms,
186186
group_freq=self.group_freq,
@@ -202,7 +202,7 @@ def setup(self, stage=None):
202202
return_parcels=self.return_parcels
203203
)
204204

205-
self.dataset_eval = PatchesDataset(root_path_coco=self.root_path_coco,
205+
self.dataset_eval = PADDataset(root_path_coco=self.root_path_coco,
206206
coco=coco_val,
207207
group_freq=self.group_freq,
208208
compression=self.compression,
@@ -227,7 +227,7 @@ def setup(self, stage=None):
227227
# Setup datasets for testing
228228
coco_test = COCO(self.path_test)
229229

230-
self.dataset_test = PatchesDataset(root_path_coco=self.root_path_coco,
230+
self.dataset_test = PADDataset(root_path_coco=self.root_path_coco,
231231
coco=coco_test,
232232
group_freq=self.group_freq,
233233
compression=self.compression,

0 commit comments

Comments
 (0)