|
@@ -33,10 +33,11 @@ from super_gradients.training.dataloaders.dataloader_factory import (
|
|
pascal_voc_segmentation_train,
|
|
pascal_voc_segmentation_train,
|
|
pascal_voc_segmentation_val,
|
|
pascal_voc_segmentation_val,
|
|
supervisely_persons_train,
|
|
supervisely_persons_train,
|
|
- supervisely_persons_val
|
|
|
|
|
|
+ supervisely_persons_val, pascal_voc_detection_train, pascal_voc_detection_val
|
|
)
|
|
)
|
|
-from super_gradients.training.datasets import COCODetectionDataset, ImageNetDataset, PascalAUG2012SegmentationDataSet, PascalVOC2012SegmentationDataSet, \
|
|
|
|
- SuperviselyPersonsDataset
|
|
|
|
|
|
+from super_gradients.training.datasets import COCODetectionDataset, ImageNetDataset, PascalAUG2012SegmentationDataSet, \
|
|
|
|
+ PascalVOC2012SegmentationDataSet, \
|
|
|
|
+ SuperviselyPersonsDataset, PascalVOCDetectionDataset
|
|
|
|
|
|
|
|
|
|
class DataLoaderFactoryTest(unittest.TestCase):
|
|
class DataLoaderFactoryTest(unittest.TestCase):
|
|
@@ -197,6 +198,16 @@ class DataLoaderFactoryTest(unittest.TestCase):
|
|
self.assertTrue(isinstance(dl, DataLoader))
|
|
self.assertTrue(isinstance(dl, DataLoader))
|
|
self.assertTrue(isinstance(dl.dataset, SuperviselyPersonsDataset))
|
|
self.assertTrue(isinstance(dl.dataset, SuperviselyPersonsDataset))
|
|
|
|
|
|
|
|
+ def test_pascal_voc_train_creation(self):
|
|
|
|
+ dl = pascal_voc_detection_train()
|
|
|
|
+ self.assertTrue(isinstance(dl, DataLoader))
|
|
|
|
+ self.assertTrue(isinstance(dl.dataset, PascalVOCDetectionDataset))
|
|
|
|
+
|
|
|
|
+ def test_pascal_voc_val_creation(self):
|
|
|
|
+ dl = pascal_voc_detection_val()
|
|
|
|
+ self.assertTrue(isinstance(dl, DataLoader))
|
|
|
|
+ self.assertTrue(isinstance(dl.dataset, PascalVOCDetectionDataset))
|
|
|
|
+
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
unittest.main()
|