def _get_dataset(self):
dic = {}
dic["file_path"] = "%s" % self.hyper["file_path"]
dic["label_size"] = 1
dic["class_nums"] = 10
if self.hyper["dataset"] == "mnist":
dic["img_size"] = 28 * 28
dataset = mnist_dataset.MnistSet(dic)
if self.hyper["dataset"] == "cifar10":
dic["img_size"] = 32 * 32 * 3
dataset = cifar10_dataset.Cifar10Set(dic)
return dataset
def test_generator_mnist_test_images():
dic = get_mnist_dic()
test_img_path = "./mnist_img/test/"
dataset = mnist_dataset.MnistSet(dic)
dataset.generator_test_images(test_img_path)