def test_no_shuffle_limit(self):
d = Dataset('data/')
d = d.include('*.png')
data = d.compile()
ld = Loader(data, data, threads=4)
ld.cropper(RandomCrop(1))
ld.image_augmentation()
itr = ld.make_one_shot_iterator([4, 3, 16, 16], 10, False,
data.capacity / 2)
ret = list(itr)
self.assertEqual(len(ret), 10)
self.assert_psnr(ret)
itr = ld.make_one_shot_iterator([4, 3, 16, 16], 10, False,
data.capacity / 2)
ret = list(itr)
self.assertEqual(len(ret), 10)
self.assert_psnr(ret)
def test_memory_limit():
d = Dataset('data/')
d = d.include('*.png')
data = d.compile()
ld = Loader(data, data, threads=4)
ld.cropper(RandomCrop(1))
ld.image_augmentation()
itr = ld.make_one_shot_iterator([4, 3, 16, 16], 10, True,
data.capacity / 2)
ret = list(itr)
assert len(ret) is 10
assert_psnr(ret)
itr = ld.make_one_shot_iterator([4, 3, 16, 16], 10, True,
data.capacity / 2)
ret = list(itr)
assert len(ret) is 10
assert_psnr(ret)