#load saved training data in cropped dimensions directly print('loading train volumes') train_imgs, train_labels = dt.load_cropped_img_labels(train_list) #print('train shape',train_imgs.shape,train_labels.shape) #load validation volumes id numbers to save the best model during training val_list = data_list.val_data(parse_config.no_of_tr_imgs, parse_config.comb_tr_imgs) #load val data both in original dimensions and its cropped dimensions print('loading val volumes') val_label_orig, val_img_crop, val_label_crop, pixel_val_list = load_val_imgs( val_list, dt, orig_img_dt) # get test volumes id list print('get test volumes list') test_list = data_list.test_data() ###################################### ###################################### #define directory to save the model save_dir = str(cfg.srt_dir) + '/models/' + str( parse_config.dataset) + '/trained_models/train_baseline/' save_dir = str(save_dir) + '/with_data_aug/' if (parse_config.rd_en == 1 and parse_config.ri_en == 1): save_dir = str(save_dir) + 'rand_deforms_and_ints_en/' elif (parse_config.rd_en == 1): save_dir = str(save_dir) + 'rand_deforms_en/' elif (parse_config.ri_en == 1): save_dir = str(save_dir) + 'rand_ints_en/'
#load saved training data in cropped dimensions directly print('load train volumes') train_imgs, train_labels = dt.load_cropped_img_labels(train_list) #print('train shape',train_imgs.shape,train_labels.shape) #load validation volumes id numbers to save the best model during training val_list = data_list.val_data(parse_config.no_of_tr_imgs, parse_config.comb_tr_imgs) #load val data both in original dimensions and its cropped dimensions print('load val volumes') val_label_orig, val_img_crop, val_label_crop, pixel_val_list = load_val_imgs( val_list, dt, orig_img_dt) # get test volumes id list print('get test volumes list') test_list = data_list.test_data(parse_config.no_of_tr_imgs) ###################################### ###################################### # parameters values set for training of CNN mean_f1_val_prev = 0.0000001 threshold_f1 = 0.0000001 step_val = parse_config.n_iter start_epoch = 0 n_epochs = step_val tr_loss_list, val_loss_list = [], [] tr_dsc_list, val_dsc_list = [], [] ep_no_list = [] loss_least_val = 1 f1_mean_least_val = 0.0000000001