Commit 78c0fa20 authored by Pietro Morerio's avatar Pietro Morerio
Browse files

override last commit

parent 1cf3dca3
This diff is collapsed.
......@@ -17,7 +17,8 @@ def show_batch(batch):
def add_defect(image):
# pepper noise only
image = random_noise(image, mode='pepper')
image = random_noise(image, mode='pepper', seed=1234)
#~ image = random_noise(image, mode='speckle', mean=-0.1, var = 0.01)
return image
......@@ -29,7 +30,7 @@ def main():
len_test = len(mnist.test.images)
len_train = len(mnist.train.images)
batch=train_images[10:30]
batch=train_images[10:20]
for i, image in enumerate(batch):
batch[i]=add_defect(image)
......
......@@ -7,7 +7,7 @@ from dataset import BatchGenerator, MyBatchGenerator, get_mnist
from model import *
flags.DEFINE_integer('batch_size', 128, 'Batch size.')
flags.DEFINE_integer('train_iter', 50000, 'Total training iter')
flags.DEFINE_integer('train_iter', 30000, 'Total training iter')
flags.DEFINE_integer('step', 500, 'Save after ... iteration')
mnist = get_mnist()
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment