Example #1
0
 def test_color_mode_setvalue(self):
     self.cp['DATASET'] = {
         "data_entry": "image/data_entry.csv",
         "class_names": "",
         "output_dir": ""
     }
     self.cp['IMAGE'] = {"image_dir": "image/", "color_mode": "rgb"}
     ic = ImageConfig(self.cp)
     self.assertEqual(ic.color_mode, "rgb")
     self.cp['IMAGE'] = {"image_dir": "image/", "color_mode": "grayscale"}
     ic = ImageConfig(self.cp)
     self.assertEqual(ic.color_mode, "grayscale")
     self.cp['IMAGE'] = {"image_dir": "image/", "color_mode": "hsv"}
     ic = ImageConfig(self.cp)
     self.assertEqual(ic.color_mode, "hsv")
Example #2
0
 def test_image_dimension_setvalue(self):
     self.cp['DATASET'] = {
         "data_entry": "image/data_entry.csv",
         "class_names": "",
         "output_dir": ""
     }
     self.cp['IMAGE'] = {"image_dir": "image/", "image_dimension": 1024}
     ic = ImageConfig(self.cp)
     self.assertEqual(ic.img_dim, 1024)
     self.cp['IMAGE'] = {"image_dir": "image/", "image_dimension": 512}
     ic = ImageConfig(self.cp)
     self.assertEqual(ic.img_dim, 512)
     self.cp['IMAGE'] = {"image_dir": "image/", "image_dimension": 256}
     ic = ImageConfig(self.cp)
     self.assertEqual(ic.img_dim, 256)
Example #3
0
 def test_scale_setvalue(self):
     self.cp['DATASET'] = {
         "data_entry": "image/data_entry.csv",
         "class_names": "",
         "output_dir": ""
     }
     self.cp['IMAGE'] = {"image_dir": "image/", "scale": 1.0}
     ic = ImageConfig(self.cp)
     self.assertEqual(ic.scale, 1.0)
     self.cp['IMAGE'] = {"image_dir": "image/", "scale": 0.5}
     ic = ImageConfig(self.cp)
     self.assertEqual(ic.scale, 0.5)
     self.cp['IMAGE'] = {"image_dir": "image/", "scale": 0.1}
     ic = ImageConfig(self.cp)
     self.assertEqual(ic.scale, 0.1)
Example #4
0
 def test_image_dimension_default(self):
     self.cp['DATASET'] = {
         "data_entry": "image/data_entry.csv",
         "class_names": "",
         "output_dir": ""
     }
     self.cp['IMAGE'] = {"image_dir": "image/"}
     ic = ImageConfig(self.cp)
     self.assertEqual(ic.img_dim, 256)
Example #5
0
 def test_scale_default(self):
     self.cp['DATASET'] = {
         "data_entry": "image/data_entry.csv",
         "class_names": "",
         "output_dir": ""
     }
     self.cp['IMAGE'] = {"image_dir": "image/"}
     ic = ImageConfig(self.cp)
     self.assertEqual(ic.scale, 1. / 255)
Example #6
0
 def test_color_mode_type(self):
     ic = ImageConfig(self.cp)
     self.assertIsInstance(ic.color_mode, str)
Example #7
0
 def test_scale_type(self):
     ic = ImageConfig(self.cp)
     self.assertIsInstance(ic.scale, float)
Example #8
0
 def test_image_dimension_type(self):
     ic = ImageConfig(self.cp)
     self.assertIsInstance(ic.img_dim, int)
 def test_ImageConfigEqual(self):
     ic = ImageConfig(self.cp)
     dc = DatasetConfig(self.cp)
     self.assertEqual(dc.ImageConfig, ic)
Example #10
0
 def ImageConfig(self):
     return ImageConfig(self.cp)