Commit b5b52c12 authored by Christoph Sommer's avatar Christoph Sommer

shuffle inital patches befor validation patch extraction

parent 23fe0131
......@@ -297,6 +297,8 @@ def train_predict(n_tiles=(1,4,4), params=params, **unet_config):
patches = N2V_DataGenerator().generate_patches_from_list(img_ch, num_patches_per_img=npatches, shape=params['patch_size'], augment=params['augment'])
numpy.random.shuffle(patches)
sep = int(len(patches)*0.9)
X = patches[:sep]
X_val = patches[ sep:]
......
Markdown is supported
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