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)