提交 a1176e55 编写于 作者: Y Yan Chunwei 提交者: daminglu

Feature/fix online scratch (#160)

上级 9a5a3bc9
...@@ -32,24 +32,34 @@ for step in range(1, 100): ...@@ -32,24 +32,34 @@ for step in range(1, 100):
# create histogram # create histogram
with logw.mode('train') as logger: with logw.mode('train') as logger:
histogram = logger.histogram("scratch/histogram", num_buckets=200) histogram = logger.histogram("scratch/histogram", num_buckets=200)
histogram0 = logger.histogram("scratch/histogram0", num_buckets=200)
for step in range(1, 100): for step in range(1, 100):
histogram.add_record(step, histogram.add_record(step,
np.random.normal( np.random.normal(
0.1 + step * 0.001, 0.1 + step * 0.001,
200. / (100 + step), 200. / (100 + step),
size=1000)) size=1000))
for step in range(1, 50):
histogram0.add_record(step,
np.random.normal(
0.1 + step * 0.003,
200. / (120 + step),
size=1000))
# create image # create image
with logw.mode("train") as logger: with logw.mode("train") as logger:
image = logger.image("scratch/dog", 4, image = logger.image("scratch/dog", 4) # randomly sample 4 images one pass
1) # randomly sample 4 images one pass image0 = logger.image("scratch/random", 4)
dog_jpg = Image.open(os.path.join(ROOT, 'python/dog.jpg')) dog_jpg = Image.open(os.path.join(ROOT, 'python/dog.jpg'))
dog_jpg = dog_jpg.resize(np.array(dog_jpg.size) / 2) dog_jpg = dog_jpg.resize(np.array(dog_jpg.size) / 2)
shape = [dog_jpg.size[1], dog_jpg.size[0], 3] shape = [dog_jpg.size[1], dog_jpg.size[0], 3]
# add dog's image
for pass_ in xrange(4): for pass_ in xrange(4):
image.start_sampling() image.start_sampling()
for sample in xrange(10): for sample in xrange(10):
# randomly crop a demo image. # randomly crop a dog's image.
target_shape = [100, 100, 3] # width, height, channels(3 for RGB) target_shape = [100, 100, 3] # width, height, channels(3 for RGB)
left_x = random.randint(0, shape[1] - target_shape[1]) left_x = random.randint(0, shape[1] - target_shape[1])
left_y = random.randint(0, shape[0] - target_shape[0]) left_y = random.randint(0, shape[0] - target_shape[0])
...@@ -57,7 +67,8 @@ with logw.mode("train") as logger: ...@@ -57,7 +67,8 @@ with logw.mode("train") as logger:
right_y = left_y + target_shape[0] right_y = left_y + target_shape[0]
# a more efficient way to sample images # a more efficient way to sample images
idx = image.is_sample_taken() # check whether this image will be taken by reservoir sampling # check whether this image will be taken by reservoir sampling
idx = image.is_sample_taken()
if idx >= 0: if idx >= 0:
data = np.array( data = np.array(
dog_jpg.crop((left_x, left_y, right_x, dog_jpg.crop((left_x, left_y, right_x,
...@@ -72,3 +83,12 @@ with logw.mode("train") as logger: ...@@ -72,3 +83,12 @@ with logw.mode("train") as logger:
# image.add_sample(shape, data) # image.add_sample(shape, data)
image.finish_sampling() image.finish_sampling()
# add randomly generated image
for pass_ in xrange(4):
image0.start_sampling()
for sample in xrange(10):
shape = [40, 30, 3]
data = np.random.random(shape).flatten()
image0.add_sample(shape, list(data))
image0.finish_sampling()
...@@ -109,14 +109,13 @@ def index(): ...@@ -109,14 +109,13 @@ def index():
@app.route('/static/<path:filename>') @app.route('/static/<path:filename>')
def serve_static(filename): def serve_static(filename):
print 'serve static ', filename
return send_from_directory( return send_from_directory(
os.path.join(server_path, static_file_path), filename) os.path.join(server_path, static_file_path), filename)
@app.route('/graphs/image') @app.route('/graphs/image')
def serve_graph(): def serve_graph():
return send_file(graph_image_path) return send_file(os.path.join(os.getcwd(), graph_image_path))
@app.route('/data/logdir') @app.route('/data/logdir')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册