def test_DATAAUGMENTATION_parameter_classification(self):
data_aug = Data_Augmentation()
data_aug.config_p_per_sample = 1
data_aug.seg_augmentation = False
img_aug, seg_aug = data_aug.run(self.img3D, self.seg3D)
self.assertFalse(np.array_equal(img_aug, self.img3D))
self.assertTrue(np.array_equal(seg_aug, self.seg3D))
def test_DATAAUGMENTATION_BASE_create(self):
data_Aug = Data_Augmentation()
self.assertIsInstance(data_Aug, Data_Augmentation)
Data_Augmentation(cycles=5)
Data_Augmentation(cycles=1,
scaling=True,
rotations=True,
elastic_deform=False,
mirror=False,
brightness=True,
contrast=True,
gamma=True,
gaussian_noise=True)
def test_DATAAUGMENTATION_parameter_gaussiannoise(self):
data_aug = Data_Augmentation(cycles=1,
scaling=False,
rotations=False,
elastic_deform=False,
mirror=False,
brightness=False,
contrast=False,
gamma=False,
gaussian_noise=True)
data_aug.config_p_per_sample = 1
img_aug, seg_aug = data_aug.run(self.img3D, self.seg3D)
self.assertFalse(np.array_equal(img_aug, self.img3D))
self.assertTrue(np.array_equal(seg_aug, self.seg3D))
def test_DATAAUGMENTATION_parameter_cropping(self):
data_aug = Data_Augmentation(cycles=1,
scaling=False,
rotations=False,
elastic_deform=False,
mirror=False,
brightness=False,
contrast=False,
gamma=False,
gaussian_noise=False)
data_aug.cropping = True
data_aug.cropping_patch_shape = (4, 4, 4)
img_aug, seg_aug = data_aug.run(self.img3D, self.seg3D)
self.assertEqual(img_aug.shape, (1, 4, 4, 4, 1))
self.assertEqual(seg_aug.shape, (1, 4, 4, 4, 3))
def setUpClass(self):
np.random.seed(1234)
# Create imgaging and segmentation data set
self.dataset = dict()
for i in range(0, 10):
img = np.random.rand(16, 16, 16) * 255
self.img = img.astype(int)
seg = np.random.rand(16, 16, 16) * 3
self.seg = seg.astype(int)
sample = (self.img, self.seg)
self.dataset["TEST.sample_" + str(i)] = sample
# Initialize Dictionary IO Interface
io_interface = Dictionary_interface(self.dataset,
classes=3,
three_dim=True)
# Initialize temporary directory
self.tmp_dir = tempfile.TemporaryDirectory(prefix="tmp.CESP.")
tmp_batches = os.path.join(self.tmp_dir.name, "batches")
# Initialize Data IO
self.data_io = Data_IO(io_interface,
input_path="",
output_path="",
batch_path=tmp_batches,
delete_batchDir=False)
# Initialize Data Augmentation
self.data_aug = Data_Augmentation()
# Get sample list
self.sample_list = self.data_io.get_indiceslist()
def test_DATAAUGMENTATION_parameter_percentage(self):
data_aug = Data_Augmentation(cycles=100,
scaling=True,
rotations=False,
elastic_deform=False,
mirror=False,
brightness=False,
contrast=False,
gamma=False,
gaussian_noise=False)
data_aug.config_p_per_sample = 0.3
img_aug, seg_aug = data_aug.run(self.img3D, self.seg3D)
counter_equal = 0
for i in range(0, 100):
is_equal = np.array_equal(img_aug[i], self.img3D[0])
if is_equal: counter_equal += 1
ratio = counter_equal / 100
self.assertTrue(ratio >= 0.5 and ratio <= 0.9)
def test_DATAAUGMENTATION_BASE_cycles(self):
with self.assertRaises(Exception):
data_aug = Data_Augmentation(cycles=0)
img_aug, seg_aug = data_aug.run(self.img2D, self.seg2D)
for i in range(1, 50, 5):
data_aug = Data_Augmentation(cycles=i)
img_aug, seg_aug = data_aug.run(self.img2D, self.seg2D)
self.assertEqual(img_aug.shape[0], i)
def test_DATAGENERATOR_augcyling(self):
data_aug = Data_Augmentation(cycles=20)
pp_fi = Preprocessor(self.data_io,
batch_size=4,
data_aug=data_aug,
prepare_subfunctions=False,
prepare_batches=False,
analysis="fullimage")
data_gen = DataGenerator(self.sample_list,
pp_fi,
training=True,
shuffle=False,
iterations=None)
self.assertEqual(50, len(data_gen))
def test_DATAAUGMENTATION_BASE_run2D(self):
data_aug = Data_Augmentation()
data_aug.config_p_per_sample = 1
img_aug, seg_aug = data_aug.run(self.img2D, self.seg2D)
self.assertEqual(img_aug.shape, self.img2D.shape)
self.assertFalse(np.array_equal(img_aug, self.img2D))
self.assertEqual(seg_aug.shape, self.seg2D.shape)
self.assertFalse(np.array_equal(seg_aug, self.seg2D))
data_aug = Data_Augmentation(cycles=1,
scaling=False,
rotations=False,
elastic_deform=False,
mirror=False,
brightness=False,
contrast=False,
gamma=False,
gaussian_noise=False)
img_aug, seg_aug = data_aug.run(self.img2D, self.seg2D)
self.assertTrue(np.array_equal(img_aug, self.img2D))