more training on 3 class problem, now also with other boundary widths

kirayuta · May 17, 2017 · 7222650 · 7222650
1 parent 391eba7
commit 7222650
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 14 deletions.
diff --git a/code/predict_generator.py b/code/predict_generator.py
@@ -1,21 +1,19 @@
+import helper.visualize
 import helper.data_provider
 import helper.model_builder
-import helper.visualize
 
 import skimage.io
 
-import matplotlib.pyplot as plt
-
 import sys
 
 data_dir_x = "/home/jr0th/github/segmentation/data/BBBC022_hand_200/test/x/"
-data_dir_y = "/home/jr0th/github/segmentation/data/BBBC022_hand_200/test/y_label_binary/"
+data_dir_y = "/home/jr0th/github/segmentation/data/BBBC022_hand_200/test/y_label_binary_2/"
 
 
 out_label = 'pred_generator'
-out_dir = '/home/jr0th/github/segmentation/out_3class_random/'
-# data_dir = '/home/jr0th/github/segmentation/data/BBBC022/'
-weights_path = '/home/jr0th/github/segmentation/checkpoints/checkpoint_3class_random.hdf5'
+out_dir = '/home/jr0th/github/segmentation/out_3class_random_2/'
+
+weights_path = '/home/jr0th/github/segmentation/checkpoints/checkpoint_3class_random_2.hdf5'
 batch_size = 10
 bit_depth = 8
 

diff --git a/code/training-hand-200.py b/code/training-hand-200.py
@@ -26,17 +26,17 @@
 # constants
 const_lr = 1e-4
 
-out_dir = "../out_3class_random_4/"
-tb_log_dir = "../logs/logs_tensorboard_3class_random_4/"
+out_dir = "../out_3class_random_8/"
+tb_log_dir = "../logs/logs_tensorboard_3class_random_8/"
 
-out_dir_log = "../logs/log_3class_random_4.csv"
-checkpoint_path = "../checkpoints/checkpoint_3class_random_4.hdf5"
+out_dir_log = "../logs/log_3class_random_8.csv"
+checkpoint_path = "../checkpoints/checkpoint_3class_random_8.hdf5"
 
 train_dir_x = '/home/jr0th/github/segmentation/data/BBBC022_hand_200/training/x_big/'
-train_dir_y = '/home/jr0th/github/segmentation/data/BBBC022_hand_200/training/y_big_label_binary_4/'
+train_dir_y = '/home/jr0th/github/segmentation/data/BBBC022_hand_200/training/y_big_label_binary_8/'
 
 val_dir_x = '/home/jr0th/github/segmentation/data/BBBC022_hand_200/validation/x'
-val_dir_y = '/home/jr0th/github/segmentation/data/BBBC022_hand_200/validation/y_label_binary_4'
+val_dir_y = '/home/jr0th/github/segmentation/data/BBBC022_hand_200/validation/y_label_binary_8'
 
 data_type = "images" # "images" or "array"
 
@@ -56,7 +56,7 @@
 # build session running on GPU 1
 configuration = tf.ConfigProto()
 configuration.gpu_options.allow_growth = True
-configuration.gpu_options.visible_device_list = "0"
+configuration.gpu_options.visible_device_list = "2"
 session = tf.Session(config = configuration)
 
 # apply session