def preprocessImage(pool, image_files, g_conf): futures = [] temp_dir = g_conf["temp_dir"] start_y = g_conf["start_y"] end_y = g_conf["end_y"] resize = g_conf["resize"] preproces_dir = create_specified_dir(temp_dir, "1_preprocess") for path in image_files: futures.append( pool.apply_async(preprocess_image, (path, temp_dir, start_y, end_y, resize))) imageObjs = [] for future in futures: img, path = future.get() imageObjs.append(img) new_file_name = get_specified_dir(preproces_dir, get_png_name_for_jpeg(path)) cv2.imwrite(new_file_name, img) return preproces_dir
def get_mask_file_name(img_file): data_dir = get_data_dir(img_file) masks_dir = get_specified_dir(data_dir, "2_masks") png_name = get_png_name_for_jpeg(img_file) return os.path.join(masks_dir, png_name)
def get_preprocess_img_name(img_file): ds_dir = os.path.join(get_data_dir(img_file), "1_preprocess") return os.path.join(ds_dir, get_png_name_for_jpeg(img_file))
def get_seg_file_name(img_file): data_dir = get_data_dir(img_file) seg_dir = get_specified_dir(data_dir, "3_segmentation") png_name = get_png_name_for_jpeg(img_file) return os.path.join(seg_dir, png_name)