1717class 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