def do_experiment():
ROOT_RESULT = "/Users/popunbom/Google Drive/情報学部/研究/修士/最終発表/Thesis/img/result"
ROOT_ROAD_MASK = "img/resource/road_mask"
ROOT_VEG_MASK = "img/resource/vegetation_mask"
for exp_num in [1, 2, 3, 5, 9]:
result = imread_with_error(
join(ROOT_RESULT, f"aerial_roi{exp_num}/result.png"))
path_veg_mask = join(ROOT_VEG_MASK, f"aerial_roi{exp_num}.png")
vegetation_mask = None
if exists(path_veg_mask):
vegetation_mask = imread_with_error(
path_veg_mask, cv2.IMREAD_GRAYSCALE).astype(bool)
road_mask = imread_with_error(
join(ROOT_ROAD_MASK, f"aerial_roi{exp_num}.png"),
cv2.IMREAD_GRAYSCALE).astype(bool)
result = ~np.all(result == 0, axis=2)
logger = ImageLogger(base_path="./tmp/detect_road_damage_v2",
prefix=f"aerial_roi{exp_num}")
# detect_road_damage_1(result, road_mask, logger)
detect_road_damage_2(result, road_mask, logger=logger)
if vegetation_mask is not None:
detect_road_damage_2(result,
road_mask,
vegetation_mask=vegetation_mask,
logger=logger)
def calc_hsv_mean_each_image():
ROOT_DIR_SRC = "./img/resource/aerial_image/fixed_histogram_v2"
ROOT_DIR_GT = "./img/resource/ground_truth"
# experiments = [1, 2, 3, 4, 5, 6]
experiments = [5]
params_mean_shift = {
# "spatial_radius": 8,
# "range_radius": 5,
"spatial_radius": 8,
"range_radius": 5,
"min_density": 0
}
gt_type = "GT_ORANGE"
results = dict()
for exp_num in experiments:
src_img = imread_with_error(
path.join(ROOT_DIR_SRC, f"aerial_roi{exp_num}.png"))
ground_truth = imread_with_error(
path.join(ROOT_DIR_GT, f"aerial_roi{exp_num}.png"))
eprint(
dedent(f"""
Experiment Num: {exp_num}
gt_type: {gt_type}
"""))
eprint(f"Do Mean-Shift ... ", end="")
src_img = pymeanshift.segment(src_img, **params_mean_shift)[0]
eprint("done")
metrics = calc_hsv_metrics_by_ground_truth(src_img, ground_truth)
# print(dedent(f"""
# Mean (H): {means['H']}
# Mean (S): {means['S']}
# Mean (V): {means['V']}
# """))
results[f"aerial_roi{exp_num}"] = metrics
for exp_name, metrics in results.items():
print("\t".join(["EXP_NAME", exp_name]))
print("\t".join(["Ch.", *list(metrics.values())[0].keys()]))
for ch_name, metric in metrics.items():
# for k, v in results.items():
# print(",".join([
# k,
# *metric.values()
# ]))
print("\t".join([str(x) for x in [ch_name, *metric.values()]]))
return results
def remove_tiny_areas_and_recalc_score():
RESULT_ROOT_DIR = "tmp/detect_building_damage/master-v2/fixed_histogram"
RESULT_GROUND_TRUTH = "img/resource/ground_truth"
C_RED = [0, 0, 255]
C_ORANGE = [0, 127, 255]
target_dirs = [
dir_path for (dir_path, dir_names, file_names) in walk(RESULT_ROOT_DIR)
if dir_path.endswith("meanshift_and_color_thresholding")
]
for target_dir in target_dirs:
eprint(f"Target: {target_dir}")
gt_type, experiment_num = re.match(r".*/GT_(.*)/aerial_roi([0-9]).*",
target_dir).groups()
eprint(
dedent(f"""
Experiment Num: {experiment_num}
GT_TYPE: GT_{gt_type}
"""))
result = imread_with_error(
path.join(target_dir, "building_damage.tiff"),
cv2.IMREAD_UNCHANGED)
# Fixing Image
result_fixed = BuildingDamageExtractor._remove_tiny_area(
(result * 255.0).astype(np.uint8))
imwrite_with_error(path.join(target_dir, "building_damage_fixed.tiff"),
result_fixed.astype(np.float32))
# Re-calc Score
ground_truth = imread_with_error(
path.join(RESULT_GROUND_TRUTH, f"aerial_roi{experiment_num}.png"))
if gt_type == "BOTH":
ground_truth = np.all(
(ground_truth == C_RED) | (ground_truth == C_ORANGE), axis=2)
elif gt_type == "RED":
ground_truth = np.all(ground_truth == C_RED, axis=2)
elif gt_type == "ORANGE":
ground_truth = np.all(ground_truth == C_ORANGE, axis=2)
cm, scores = evaluation_by_confusion_matrix(result_fixed, ground_truth)
j = {"Confusion Matrix": cm, "Score": scores}
json.dump(j,
open(path.join(target_dir, "scores_fixed_result.json"), "w"),
ensure_ascii=False,
sort_keys=True,
indent="\t")
def gen_edge_images():
IMG_PATH = "img/resource/aerial_image/aerial_roi1.png"
SAVE_DIR = "/Users/popunbom/Google Drive/情報学部/研究/修士/最終発表/Thesis/figs"
# ROI: X, Y, W, H
X, Y, W, H = 190, 140, 200, 200
img = imread_with_error(IMG_PATH)
inst = EdgeProcedures(img)
G = inst.edge_magnitude
G = (cm.get_cmap("jet")(G) * 255).astype(np.uint8)[:, :, [2, 1, 0]]
A = inst.get_angle_colorized_img(max_intensity=True)
A_with_magnitude = inst.get_angle_colorized_img()
print(G.shape, G.dtype, G.min(), G.max())
print(A.shape, A.dtype, A.min(), A.max())
img = img[Y:Y + H, X:X + W, :]
G = G[Y:Y + H, X:X + W, :]
A = A[Y:Y + H, X:X + W, :]
A_with_magnitude = A_with_magnitude[Y:Y + H, X:X + W, :]
write_images(
SAVE_DIR,
[
("edge_input", img),
("edge_magnitude", G),
("edge_angle", A),
("edge_angle_with_magnitude", A_with_magnitude)
],
prefix=""
)
def generate_scores_each_gt_type():
RESULT_ROOT_DIR = "tmp/detect_building_damage/master-v3/fixed_histogram/GT_BOTH"
RESULT_GROUND_TRUTH = "img/resource/ground_truth"
MAP_RESULT_IMAGE = {
"meanshift_and_color_thresholding": "building_damage_fixed.tiff",
"edge_angle_variance_with_hpf": "building_damage.tiff",
"edge_pixel_classify": "building_damage.tiff"
}
C_RED = [0, 0, 255]
C_ORANGE = [0, 127, 255]
target_dirs = [
path.join(RESULT_ROOT_DIR, entry) for entry in listdir(RESULT_ROOT_DIR)
if isdir(path.join(RESULT_ROOT_DIR, entry))
]
for target_dir in target_dirs:
eprint(f"Target: {target_dir}")
experiment_num, method_name = re.match(
r".*/aerial_roi([0-9])_[0-9]{8}_[0-9]{6}_(.*)$",
target_dir).groups()
eprint(
dedent(f"""
Experiment Num: {experiment_num}
Method: {method_name}
"""))
GT = cv2.imread(
path.join(RESULT_GROUND_TRUTH, f"aerial_roi{experiment_num}.png"))
result = imread_with_error(
path.join(target_dir, MAP_RESULT_IMAGE[method_name]),
cv2.IMREAD_UNCHANGED).astype(bool)
for gt_type in ["GT_BOTH", "GT_RED", "GT_ORANGE"]:
if gt_type == "GT_BOTH":
ground_truth = np.all((GT == C_RED) | (GT == C_ORANGE), axis=2)
elif gt_type == "GT_RED":
ground_truth = np.all(GT == C_RED, axis=2)
elif gt_type == "GT_ORANGE":
ground_truth = np.all(GT == C_ORANGE, axis=2)
cm, scores = evaluation_by_confusion_matrix(result, ground_truth)
j = {"Confusion Matrix": cm, "Score": scores}
save_path = path.join(target_dir, "evaluation", gt_type)
if not exists(save_path):
makedirs(save_path)
json.dump(j,
open(path.join(save_path, "scores.json"), "w"),
ensure_ascii=False,
sort_keys=True,
indent="\t")
def gen_source_overlay():
ROOT_DIR = "/Users/popunbom/Google Drive/情報学部/研究/修士/最終発表/Thesis/img/resource"
SRC_DIR = join(ROOT_DIR, "aerial_image")
GT_DIR = join(ROOT_DIR, "ground_truth")
ROAD_MASK_DIR = join(ROOT_DIR, "road_mask")
for exp_num in [1, 2, 3, 5, 9]:
eprint("Experiment Num:", exp_num)
bg, fg, road_mask = [
imread_with_error(
join(root_dir, f"aerial_roi{exp_num}.png")
)
for root_dir in [
SRC_DIR,
GT_DIR,
ROAD_MASK_DIR
]
]
road_mask[:, :, [0, 2]] = [0, 0]
gt_overlay = hsv_blending(bg, fg)
imwrite_with_error(
join(
GT_DIR,
f"aerial_roi{exp_num}_overlay.png"
),
gt_overlay
)
road_mask_overlay = hsv_blending(bg, road_mask)
imwrite_with_error(
join(
ROAD_MASK_DIR,
f"aerial_roi{exp_num}_overlay.png"
),
road_mask_overlay
)
def find_ms_params(n):
file_name = f"aerial_roi{n}.png"
src = imread_with_error(
join(ROOT_DIR_SRC, file_name)
)
ans = imread_with_error(
join(ROOT_DIR_ANS, file_name)
)
ms_params = sum([
[
{
"spatial_radius": sp,
"range_radius": sr,
"min_density": 0
}
for sr in np.arange(SR_RANGE[0], SR_RANGE[0]+SR_RANGE[1], SR_RANGE[2])
]
for sp in np.arange(SP_RANGE[0], SP_RANGE[0]+SP_RANGE[1], SP_RANGE[2])
], [])
progress_bar = tqdm(total=len(ms_params), position=0)
def _update_progressbar(arg):
progress_bar.update()
cp = CustomPool()
pool = cp.Pool(n_process=6, initializer=tqdm.set_lock, initargs=(tqdm.get_lock(),))
results = list()
for params in ms_params:
results.append(
pool.apply_async(
func_worker,
args=(src, ),
kwds=params,
callback=_update_progressbar
)
)
pool.close()
pool.join()
cp.update()
results = [result.get() for result in results]
results = sorted(
[
(
sp,
sr,
np.sum(
np.abs(segmented - ans)
)
)
for segmented, sp, sr in results
],
key=lambda e: e[0]
)
pprint(results)
with open(f"tmp/find_ms_params_{n}.csv", "wt") as f:
f.write("spatial_radius, range_radius, n_diffs\n")
for result in results:
f.write(", ".join([ str(x) for x in result]) + "\n")
return results
def gen_result_road_damage():
ROOT_DIR_RESULT = "/Users/popunbom/.tmp/EQDmgAnalyzr/detect_road_damage_v2"
dirs = [
join(ROOT_DIR_RESULT, d)
for d in listdir(ROOT_DIR_RESULT)
if isdir(join(ROOT_DIR_RESULT, d))
]
for d in dirs:
print(d)
exp_num, = re.match(
r".*aerial_roi([0-9]).*",
d
).groups()
src = imread_with_error(
join(
ROOT_DIR_SRC,
f"aerial_roi{exp_num}.png"
)
)
src_gs = cv2.cvtColor(
cv2.cvtColor(
src,
cv2.COLOR_BGR2GRAY
),
cv2.COLOR_GRAY2BGR
)
road_mask = imread_with_error(
join(
ROOT_DIR_ROAD_MASK,
f"aerial_roi{exp_num}.png"
),
cv2.IMREAD_GRAYSCALE
).astype(bool)
src[road_mask == False] = [0, 0, 0]
result_paths = [d]
if exists(join(d, "removed_vegetation")):
result_paths.append(
join(d, "removed_vegetation")
)
for result_path in result_paths:
result = imread_with_error(
join(
result_path,
"result_extracted.tiff"
),
cv2.IMREAD_UNCHANGED
)
result = (cm.get_cmap("jet")(
result / result.max()
) * 255).astype(np.uint8)[:, :, [2, 1, 0]]
dst = hsv_blending(src, result)
dst[road_mask == False] = src_gs[road_mask == False]
imwrite_with_error(
join(
result_path,
f"study_overlay.png"
),
dst
)
def eval_by_utilize_methods():
ROOT_DIR_RESULT = "/Users/popunbom/Google Drive/情報学部/研究/修士/最終発表/Thesis/img/result"
SPLIT_PATTERNS = {
"MEAN_SHIFT" : (0,),
"ANGLE_VARIANCE" : (1,),
"PIXEL_CLASSIFY" : (2,),
"MEAN_SHIFT_AND_ANGLE_VARIANCE" : (0, 1),
"MEAN_SHIFT_AND_PIXEL_CLASSIFY" : (0, 2),
"ANGLE_VARIANCE_AND_PIXEL_CLASSIFY" : (1, 2),
"MEAN_SHIFT_AND_ANGLE_VARIANCE_AND_PIXEL_CLASSIFY": (0, 1, 2)
}
result_paths = sorted([
join(dir_path, "result.png")
for (dir_path, dir_names, file_names) in walk(ROOT_DIR_RESULT)
if "result.png" in file_names
])
for result_path in result_paths:
exp_num, = re.match(
r".*aerial_roi([0-9]).*",
result_path
).groups()
result_src = imread_with_error(
result_path
)
gt_src = imread_with_error(
join(
ROOT_DIR_GT,
f"aerial_roi{exp_num}.png"
)
)
result_dir = join(
dirname(result_path),
"evaluation"
)
if not exists(result_dir):
makedirs(result_dir)
NDARRAY_ASSERT(result_src, ndim=3)
print("File:", result_path)
print(
np.unique(
result_src.reshape(result_src.shape[0] * result_src.shape[1], result_src.shape[2]),
axis=0
)
)
# Use Vegetation Mask
OPTIONS = ["NORMAL"]
path_vegetation_mask = join(
ROOT_DIR_VEG_MASK,
f"aerial_roi{exp_num}.png"
)
vegetation_mask = None
if exists(path_vegetation_mask):
vegetation_mask = imread_with_error(
path_vegetation_mask,
cv2.IMREAD_GRAYSCALE
).astype(bool)
OPTIONS.append("REMOVED_VEGETATION")
scores = dict()
for option in OPTIONS:
for name, channels in SPLIT_PATTERNS.items():
eprint("Evaluation:", name)
result = np.zeros(result_src.shape[:2], dtype=np.int16)
for channel in channels:
result += (result_src[:, :, channel] != 0)
result = result.astype(bool)
if option == "NORMAL":
save_dir = result_dir
elif option == "REMOVED_VEGETATION":
result = result & ~vegetation_mask
save_dir = join(
result_dir,
"removed_vegetation"
)
if not exists(save_dir):
makedirs(save_dir)
imwrite_with_error(
join(save_dir, name.replace(" & ", "_and_") + ".png"),
(result * 255).astype(np.uint8)
)
scores[name] = dict()
for gt_type in ["GT_BOTH", "GT_ORANGE", "GT_RED"]:
ground_truth = None
if gt_type == "GT_BOTH":
ground_truth = np.all(
(gt_src == C_RED) | (gt_src == C_ORANGE),
axis=2
)
elif gt_type == "GT_RED":
ground_truth = np.all(
gt_src == C_RED,
axis=2
)
elif gt_type == "GT_ORANGE":
ground_truth = np.all(
gt_src == C_ORANGE,
axis=2
)
cm, metrics = evaluation_by_confusion_matrix(
result,
ground_truth
)
scores[name][gt_type] = {
"Confusion Matrix": cm,
"Score" : metrics
}
json.dump(
scores,
open(join(save_dir, "scores.json"), "w"),
ensure_ascii=False,
sort_keys=True,
indent="\t"
)
def only_overlay_image():
result_dirs = sum([
[
join(
ROOT_DIR_RESULT,
gt_type,
d
)
for d in listdir(
join(
ROOT_DIR_RESULT,
gt_type
)
)
if isdir(
join(
ROOT_DIR_RESULT,
gt_type,
d
)
)
]
# for gt_type in ["GT_BOTH", "GT_ORANGE", "GT_RED"]
for gt_type in ["GT_BOTH"]
], [])
result_dirs = sorted(result_dirs)
for result_dir in result_dirs:
gt_type, experiment_num = re.match(
r".*/GT_(.*)/aerial_roi([0-9]).*",
result_dir
).groups()
eprint(dedent(f"""
Experiment Num: {experiment_num}
GT_TYPE: GT_{gt_type}
"""))
# Load: ground_truth
ground_truth = imread_with_error(
join(
ROOT_DIR_GT,
f"aerial_roi{experiment_num}.png"
)
)
# Load: source
src = imread_with_error(
join(
ROOT_DIR_SRC,
f"aerial_roi{experiment_num}.png"
)
)
src_gs = cv2.cvtColor(
cv2.cvtColor(
src,
cv2.COLOR_BGR2GRAY
),
cv2.COLOR_GRAY2BGR
)
if gt_type == "BOTH":
ground_truth = np.all(
(ground_truth == C_RED) | (ground_truth == C_ORANGE),
axis=2
)
elif gt_type == "RED":
ground_truth = np.all(
ground_truth == C_RED,
axis=2
)
elif gt_type == "ORANGE":
ground_truth = np.all(
ground_truth == C_ORANGE,
axis=2
)
ground_truth = (ground_truth * 255).astype(np.uint8)
Z = np.zeros(
ground_truth.shape[:2],
dtype=np.uint8
)
if "edge_angle_variance_with_hpf" in result_dir:
fd_angle_var = imread_with_error(
join(
result_dir,
"edge_angle_variance/angle_variance.tiff"
),
cv2.IMREAD_UNCHANGED
)
fd_angle_var = (cm.get_cmap("jet")(
fd_angle_var / fd_angle_var.max()
) * 255).astype(np.uint8)[:, :, [2, 1, 0]]
fd_hpf = imread_with_error(
join(
result_dir,
"high_pass_filter/HPF_gray.tiff"
),
cv2.IMREAD_UNCHANGED
)
fd_hpf = (cm.get_cmap("jet")(
fd_hpf / fd_hpf.max()
) * 255).astype(np.uint8)[:, :, [2, 1, 0]]
# Overlay
fd_overlay_angle_var = hsv_blending(
src_gs,
fd_angle_var
)
# fd_overlay_angle_var = overlay.do_gimp_overlay(
# src_gs, fd_angle_var, overlay.FUNC_GRAIN_MERGE
# )
fd_overlay_hpf = hsv_blending(
src_gs,
fd_hpf
)
# fd_overlay_hpf = overlay.do_gimp_overlay(
# src_gs, fd_hpf, overlay.FUNC_GRAIN_MERGE
# )
write_images(
result_dir,
[
("fd_overlay_angle_var", fd_overlay_angle_var),
("fd_overlay_hpf", fd_overlay_hpf)
]
)
elif "edge_angle_variance" in result_dir:
fd_angle_var = imread_with_error(
join(
result_dir,
"edge_angle_variance/angle_variance.tiff"
),
cv2.IMREAD_UNCHANGED
)
fd_angle_var = (cm.get_cmap("jet")(
fd_angle_var / fd_angle_var.max()
) * 255).astype(np.uint8)[:, :, [2, 1, 0]]
# Overlay
fd_overlay_angle_var = hsv_blending(
src_gs,
fd_angle_var
)
# fd_overlay_angle_var = overlay.do_gimp_overlay(
# src_gs, fd_angle_var, overlay.FUNC_GRAIN_MERGE
# )
write_images(
result_dir,
[
("fd_overlay_angle_var", fd_overlay_angle_var)
]
)
elif "edge_pixel_classify" in result_dir:
fd = imread_with_error(
join(
result_dir,
"features.tiff"
),
cv2.IMREAD_UNCHANGED
)
fd = (cm.get_cmap("jet")(
fd / fd.max()
) * 255).astype(np.uint8)[:, :, [2, 1, 0]]
# Overlay
fd_overlay = hsv_blending(
src_gs,
fd
)
# fd_overlay = overlay.do_gimp_overlay(
# src_gs, fd, overlay.FUNC_GRAIN_MERGE
# )
write_images(
result_dir,
[
("fd_overlay", fd_overlay)
]
)
def gen_study_data():
result_dirs = sum([
[
join(
ROOT_DIR_RESULT,
gt_type,
d
)
for d in listdir(
join(
ROOT_DIR_RESULT,
gt_type
)
)
if isdir(
join(
ROOT_DIR_RESULT,
gt_type,
d
)
)
]
# for gt_type in ["GT_BOTH", "GT_ORANGE", "GT_RED"]
for gt_type in ["GT_BOTH"]
], [])
result_dirs = sorted(result_dirs)
pprint(result_dirs[1:2])
for result_dir in result_dirs:
gt_type, experiment_num = re.match(
r".*/GT_(.*)/aerial_roi([0-9]).*",
result_dir
).groups()
eprint(dedent(f"""
Experiment Num: {experiment_num}
GT_TYPE: GT_{gt_type}
"""))
# Load: ground_truth
ground_truth = imread_with_error(
join(
ROOT_DIR_GT,
f"aerial_roi{experiment_num}.png"
)
)
# Load: source
src = imread_with_error(
join(
ROOT_DIR_SRC,
f"aerial_roi{experiment_num}.png"
)
)
src_gs = cv2.cvtColor(
cv2.cvtColor(
src,
cv2.COLOR_BGR2GRAY
),
cv2.COLOR_GRAY2BGR
)
if gt_type == "BOTH":
ground_truth = np.all(
(ground_truth == C_RED) | (ground_truth == C_ORANGE),
axis=2
)
elif gt_type == "RED":
ground_truth = np.all(
ground_truth == C_RED,
axis=2
)
elif gt_type == "ORANGE":
ground_truth = np.all(
ground_truth == C_ORANGE,
axis=2
)
ground_truth = (ground_truth * 255).astype(np.uint8)
Z = np.zeros(
ground_truth.shape[:2],
dtype=np.uint8
)
if "meanshift_and_color_thresholding" in result_dir:
result = (imread_with_error(
join(
result_dir,
"building_damage_fixed.tiff"
),
cv2.IMREAD_UNCHANGED
) * 255).astype(np.uint8)
confusion_matrix = np.dstack(
[result, ground_truth, result]
)
missing = confusion_matrix.copy()
missing[~np.all(missing == C_GREEN, axis=2)] = [0, 0, 0]
wrong = confusion_matrix.copy()
wrong[~np.all(wrong == C_MAGENTA, axis=2)] = [0, 0, 0]
# Extract
missing_extracted = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
src,
np.all(missing == C_GREEN, axis=2)
)
missing_extracted_with_color = hsv_blending(
src_gs,
missing
)
wrong_extracted = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
src,
np.all(wrong == C_MAGENTA, axis=2)
)
wrong_extracted_with_color = hsv_blending(
src_gs,
wrong
)
write_images(
result_dir,
[
("confusion_matrix", confusion_matrix),
("missing", missing),
("wrong", wrong),
("missing_extracted", missing_extracted),
("missing_extracted_with_color", missing_extracted_with_color),
("wrong_extracted", wrong_extracted),
("wrong_extracted_with_color", wrong_extracted_with_color)
]
)
elif "edge_angle_variance_with_hpf" in result_dir:
result = (imread_with_error(
join(
result_dir,
"building_damage.tiff"
),
cv2.IMREAD_UNCHANGED
) * 255).astype(np.uint8)
fd_angle_var = imread_with_error(
join(
result_dir,
"edge_angle_variance/angle_variance.tiff"
),
cv2.IMREAD_UNCHANGED
)
fd_angle_var = (cm.get_cmap("jet")(
fd_angle_var / fd_angle_var.max()
) * 255).astype(np.uint8)[:, :, [2, 1, 0]]
fd_hpf = imread_with_error(
join(
result_dir,
"high_pass_filter/HPF_gray.tiff"
),
cv2.IMREAD_UNCHANGED
)
fd_hpf = (cm.get_cmap("jet")(
fd_hpf / fd_hpf.max()
) * 255).astype(np.uint8)[:, :, [2, 1, 0]]
# Generate Image of Confusion Matrix
confusion_matrix = np.dstack(
[result, ground_truth, result]
)
missing = confusion_matrix.copy()
missing[~np.all(missing == C_GREEN, axis=2)] = [0, 0, 0]
wrong = confusion_matrix.copy()
wrong[~np.all(wrong == C_MAGENTA, axis=2)] = [0, 0, 0]
# Overlay
fd_overlay_angle_var = hsv_blending(
src_gs,
fd_angle_var
)
# fd_overlay_angle_var = overlay.do_gimp_overlay(
# src_gs, fd_angle_var, overlay.FUNC_GRAIN_MERGE
# )
fd_overlay_hpf = hsv_blending(
src_gs,
fd_hpf
)
# fd_overlay_hpf = overlay.do_gimp_overlay(
# src_gs, fd_hpf, overlay.FUNC_GRAIN_MERGE
# )
# Extract
missing_extracted = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
src,
np.all(missing == C_GREEN, axis=2)
)
missing_extracted_with_color = hsv_blending(
src_gs,
missing
)
missing_extracted_by_anglevar = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
fd_angle_var,
np.all(missing == C_GREEN, axis=2)
)
missing_extracted_by_hpf = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
fd_hpf,
np.all(missing == C_GREEN, axis=2)
)
wrong_extracted = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
src,
np.all(wrong == C_MAGENTA, axis=2)
)
wrong_extracted_with_color = hsv_blending(
src_gs,
wrong
)
wrong_extracted_by_anglevar = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
fd_angle_var,
np.all(wrong == C_MAGENTA, axis=2)
)
wrong_extracted_by_hpf = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
fd_hpf,
np.all(wrong == C_MAGENTA, axis=2)
)
write_images(
result_dir,
[
("fd_overlay_angle_var", fd_overlay_angle_var),
("fd_overlay_hpf", fd_overlay_hpf),
("confusion_matrix", confusion_matrix),
("missing", missing),
("wrong", wrong),
("missing_extracted", missing_extracted),
("missing_extracted_with_color", missing_extracted_with_color),
("missing_extracted_by_anglevar", missing_extracted_by_anglevar),
("missing_extracted_by_hpf", missing_extracted_by_hpf),
("wrong_extracted", wrong_extracted),
("wrong_extracted_with_color", wrong_extracted_with_color),
("wrong_extracted_by_anglevar", wrong_extracted_by_anglevar),
("wrong_extracted_by_hpf", wrong_extracted_by_hpf)
]
)
elif "edge_angle_variance" in result_dir:
result = (imread_with_error(
join(
result_dir,
"building_damage.tiff"
),
cv2.IMREAD_UNCHANGED
) * 255).astype(np.uint8)
fd_angle_var = imread_with_error(
join(
result_dir,
"edge_angle_variance/angle_variance.tiff"
),
cv2.IMREAD_UNCHANGED
)
fd_angle_var = (cm.get_cmap("jet")(
fd_angle_var / fd_angle_var.max()
) * 255).astype(np.uint8)[:, :, [2, 1, 0]]
# Generate Image of Confusion Matrix
confusion_matrix = np.dstack(
[result, ground_truth, result]
)
missing = confusion_matrix.copy()
missing[~np.all(missing == C_GREEN, axis=2)] = [0, 0, 0]
wrong = confusion_matrix.copy()
wrong[~np.all(wrong == C_MAGENTA, axis=2)] = [0, 0, 0]
# Overlay
fd_overlay_angle_var = hsv_blending(
src_gs,
fd_angle_var
)
# fd_overlay_angle_var = overlay.do_gimp_overlay(
# src_gs, fd_angle_var, overlay.FUNC_GRAIN_MERGE
# )
# Extract
missing_extracted = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
src,
np.all(missing == C_GREEN, axis=2)
)
missing_extracted_with_color = hsv_blending(
src_gs,
missing
)
missing_extracted_by_anglevar = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
fd_angle_var,
np.all(missing == C_GREEN, axis=2)
)
wrong_extracted = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
src,
np.all(wrong == C_MAGENTA, axis=2)
)
wrong_extracted_with_color = hsv_blending(
src_gs,
wrong
)
wrong_extracted_by_anglevar = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
fd_angle_var,
np.all(wrong == C_MAGENTA, axis=2)
)
write_images(
result_dir,
[
("fd_overlay_angle_var", fd_overlay_angle_var),
("confusion_matrix", confusion_matrix),
("missing", missing),
("wrong", wrong),
("missing_extracted", missing_extracted),
("missing_extracted_with_color", missing_extracted_with_color),
("missing_extracted_by_anglevar", missing_extracted_by_anglevar),
("wrong_extracted", wrong_extracted),
("wrong_extracted_with_color", wrong_extracted_with_color),
("wrong_extracted_by_anglevar", wrong_extracted_by_anglevar)
]
)
elif "edge_pixel_classify" in result_dir:
result = (imread_with_error(
join(
result_dir,
"building_damage.tiff"
),
cv2.IMREAD_UNCHANGED
) * 255).astype(np.uint8)
fd = imread_with_error(
join(
result_dir,
"features.tiff"
),
cv2.IMREAD_UNCHANGED
)
fd = (cm.get_cmap("jet")(
fd / fd.max()
) * 255).astype(np.uint8)[:, :, [2, 1, 0]]
# Overlay
fd_overlay = hsv_blending(
src_gs,
fd
)
# fd_overlay = overlay.do_gimp_overlay(
# src_gs, fd, overlay.FUNC_GRAIN_MERGE
# )
# Generate Image of Confusion Matrix
confusion_matrix = np.dstack(
[result, ground_truth, result]
)
missing = confusion_matrix.copy()
missing[~np.all(missing == C_GREEN, axis=2)] = [0, 0, 0]
wrong = confusion_matrix.copy()
wrong[~np.all(wrong == C_MAGENTA, axis=2)] = [0, 0, 0]
# Extract
missing_extracted = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
src,
np.all(missing == C_GREEN, axis=2)
)
missing_extracted_with_color = hsv_blending(
src_gs,
missing
)
missing_extracted_by_fd = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
fd,
np.all(missing == C_GREEN, axis=2)
)
wrong_extracted = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
src,
np.all(wrong == C_MAGENTA, axis=2)
)
wrong_extracted_with_color = hsv_blending(
src_gs,
wrong
)
wrong_extracted_by_fd = merge_arrays_by_mask(
(src_gs * CONT).astype(np.uint8),
fd,
np.all(wrong == C_MAGENTA, axis=2)
)
write_images(
result_dir,
[
("fd_overlay", fd_overlay),
("confusion_matrix", confusion_matrix),
("missing", missing),
("wrong", wrong),
("missing_extracted", missing_extracted),
("missing_extracted_with_color", missing_extracted_with_color),
("missing_extracted_by_fd", missing_extracted_by_fd),
("wrong_extracted", wrong_extracted),
("wrong_extracted_with_color", wrong_extracted_with_color),
("wrong_extracted_by_fd", wrong_extracted_by_fd)
]
)
def apply_vegetation_mask():
ROOT_DIR_RESULT = "/Users/popunbom/Google Drive/情報学部/研究/修士/最終発表/Thesis/img"
ROOT_DIR_VEG = "img/resource/vegetation_mask"
ROOT_DIR_SRC = join(ROOT_DIR_RESULT, "resource/aerial_image")
for exp_num in [1, 2, 3, 5, 9]:
file_name = f"aerial_roi{exp_num}.png"
result_dir = join(ROOT_DIR_RESULT, f"result/aerial_roi{exp_num}")
save_dir = join(result_dir, "extract_vegetation")
if not exists(save_dir):
makedirs(save_dir)
src = imread_with_error(
join(ROOT_DIR_SRC, file_name)
)
veg_mask = imread_with_error(
join(ROOT_DIR_VEG, file_name),
)
result = imread_with_error(
join(result_dir, "result.png")
)
imwrite_with_error(
join(
save_dir,
"thresholded.png"
),
veg_mask
)
# WHITE -> GREEN
veg_mask[:, :, [0, 2]] = [0, 0]
veg_overlay = hsv_blending(
bg_img=src,
fg_img=veg_mask,
bg_v_scale=0.6,
fg_v_scale=1.5,
)
imwrite_with_error(
join(
save_dir,
"vegetation_overlay.png"
),
veg_overlay
)
# Result - Vegetation
veg_mask_bin = veg_mask[:, :, 1].astype(bool)
result[veg_mask_bin == True] = [0, 0, 0]
imwrite_with_error(
join(
result_dir,
"result_removed_vegetation.png"
),
result
)
# Overlay result
result_overlay = hsv_blending(
bg_img=src,
fg_img=result,
bg_v_scale=0.6
)
imwrite_with_error(
join(
result_dir,
"result_overlay_removed_vegetation.png"
),
result_overlay
)
def gen_result_overlay():
ROOT_DIR = "/Users/popunbom/Google Drive/情報学部/研究/修士/最終発表/Thesis/img"
SRC_DIR = join(ROOT_DIR, "resource/aerial_image")
for exp_num in [1, 2, 3, 5, 9]:
eprint("Experiment Num:", exp_num)
bg = imread_with_error(
join(
SRC_DIR,
f"aerial_roi{exp_num}.png"
)
)
building_damage = imread_with_error(
join(
ROOT_DIR,
f"result/aerial_roi{exp_num}/result.png"
)
)
building_damage_overlay = hsv_blending(bg, building_damage, bg_v_scale=0.6)
imwrite_with_error(
join(
ROOT_DIR,
f"result/aerial_roi{exp_num}/result_overlay.png"
),
building_damage_overlay
)
road_damage = imread_with_error(
join(
ROOT_DIR,
f"result/aerial_roi{exp_num}/road_damage/thresholded.png"
)
)
# White -> Red
road_damage[:, :, [0, 1]] = [0, 0]
road_damage_overlay = hsv_blending(bg, road_damage)
imwrite_with_error(
join(
ROOT_DIR,
f"result/aerial_roi{exp_num}/road_damage/result_overlay.png"
),
road_damage_overlay
)
road_damage_wo_veg = imread_with_error(
join(
ROOT_DIR,
f"result/aerial_roi{exp_num}/road_damage/removed_vegetation/thresholded.png"
)
)
# White -> Red
road_damage_wo_veg[:, :, [0, 1]] = [0, 0]
road_damage_wo_veg_overlay = hsv_blending(bg, road_damage_wo_veg)
imwrite_with_error(
join(
ROOT_DIR,
f"result/aerial_roi{exp_num}/road_damage/removed_vegetation/result_overlay.png"
),
road_damage_wo_veg_overlay
)
def eval_road_damage():
ROOT_DIR_RESULT = "/Users/popunbom/.tmp/EQDmgAnalyzr/detect_road_damage_v2"
dirs = [
join(ROOT_DIR_RESULT, d)
for d in listdir(ROOT_DIR_RESULT)
if isdir(join(ROOT_DIR_RESULT, d))
]
for d in dirs:
print(d)
exp_num, = re.match(
r".*aerial_roi([0-9]).*",
d
).groups()
eprint(dedent(f"""
Experiment Num: {exp_num}
"""))
road_mask = imread_with_error(
join(
ROOT_DIR_ROAD_MASK,
f"aerial_roi{exp_num}.png"
),
cv2.IMREAD_GRAYSCALE
).astype(bool)
# GT: GT_BOTH
ground_truth = imread_with_error(
join(
ROOT_DIR_GT,
f"aerial_roi{exp_num}.png"
),
cv2.IMREAD_GRAYSCALE
).astype(bool)
result_dirs = [d]
if exists(join(d, "removed_vegetation")):
result_dirs.append(
join(d, "removed_vegetation")
)
for result_dir in result_dirs:
result = imread_with_error(
join(result_dir, "thresholded.png"),
cv2.IMREAD_GRAYSCALE
).astype(bool)
result = result[road_mask == True]
ground_truth_masked = ground_truth[road_mask == True]
# ground_truth[road_mask == False] = False
cm, metrics = evaluation_by_confusion_matrix(
result,
ground_truth_masked
)
result = {
"Confusion Matrix": cm,
"Score": metrics
}
json.dump(
result,
open(
join(result_dir, "scores.json"),
"w"
),
ensure_ascii=False,
sort_keys=True,
indent="\t"
)
