Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

#332 feature/SG-220_refactor_pascal_detection_dataset_interface

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-220_refactor_pascal_detection_dataset_interface
1 changed files with 14 additions and 3 deletions
  1. 14
    3
      tests/unit_tests/datalaoder_factory_test.py
@@ -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()
Discard