tmp_img = np.zeros(imshape) tmp_label = FullImage.copy() # with open('./logs/data/%s_side.pkl'%img_path,'w+') as f: # data = [tmp_img, tmp_label, img_path, ally] # pkl.dump(data, f) MSE_loss = MSE_pixel_loss(tmp_img, tmp_label, img_path, ally) print 'sideoutput:', img_path, MSE_loss MSE_loss_list.append(MSE_loss) if save_test_img: pred_img = cv2.resize(new_Image.astype(np.uint8), (ROI_img.shape[1], ROI_img.shape[0])) tmp_img = pred_img.copy() tmp_ROI_img = ROI_img.copy() pred_img[pred_img==3]=2 pred_img[NucluesImage==1]=3 ROI_img = crop_boundry(ROI_img, pred_img) alpha_img = ori_img.copy() alpha_img[BeginY:, ally[0]:ally[1]] = ROI_img ROI_img = cv2.addWeighted(alpha_img, 0.4, ori_img, 0.6, 0) ## imshow the truth ROI_img = get_truth(ROI_img, img_path, ally, BeginY) if not os.path.exists(os.path.join(args.results,model_name)): os.mkdir(os.path.join(args.results,model_name)) img_path2 = img_path[:-4] + '_ray.png' save_name = os.path.join(args.results,model_name, img_path2) cv2.imwrite(save_name, ROI_img) FullImage = np.zeros(imshape) FinalShape = FinalShape * 1 pred_imgFinal = pred_img
if epoch % 4 == 0 and i % 4 == 0 and epoch != 0: left, right, top = img_infor[path]['position'] w, h = img_infor[path]['size'][:2] new_Image = ppi.astype(np.uint8) kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3)) new_Image = cv2.morphologyEx(new_Image, cv2.MORPH_CLOSE, kernel) ori_img_path = os.path.join(ori_data, path) ori_img = cv2.imread(ori_img_path) ROI_img = ori_img[top:, left:right, :].copy() pred_img = cv2.resize(new_Image.astype(np.uint8), (ROI_img.shape[1], ROI_img.shape[0]), interpolation=cv2.INTER_LANCZOS4) ROI_img = crop_boundry(ROI_img, pred_img) alpha_img = ori_img.copy() alpha_img[top:, left:right] = ROI_img ROI_img = cv2.addWeighted(alpha_img, 0.4, ori_img, 0.6, 0) try: ROI_img = get_truth(ROI_img, path, [left, right], top) except: print path if not os.path.exists('./visual_results/%s' % model_name): os.mkdir('./visual_results/%s' % model_name) save_name = './visual_results/%s/label_%s_%s.png' % (model_name, epoch, i) cv2.imwrite(save_name, ROI_img) MSE_loss = None