|
|
import os |
|
|
import cv2 |
|
|
import numpy as np |
|
|
import subprocess |
|
|
from cfglib.config import config as cfg |
|
|
from util.misc import mkdirs |
|
|
|
|
|
|
|
|
def osmkdir(out_dir): |
|
|
import shutil |
|
|
if os.path.exists(out_dir): |
|
|
shutil.rmtree(out_dir) |
|
|
os.makedirs(out_dir) |
|
|
|
|
|
|
|
|
def analysize_result(source_dir, fid_path, outpt_dir, name): |
|
|
|
|
|
bad_txt = open("{}/eval.txt".format(outpt_dir), 'w') |
|
|
all_eval = open("{}/{}/{}_eval.txt".format(cfg.output_dir, "Analysis", name), 'a+') |
|
|
sel_list = list() |
|
|
with open(fid_path) as f: |
|
|
lines = f.read().split("\n") |
|
|
for line in lines: |
|
|
line_items = line.split(" ") |
|
|
id = line_items[0] |
|
|
precision = float(line_items[2].split('=')[-1]) |
|
|
recall = float(line_items[4].split('=')[-1]) |
|
|
if id != "ALL" and (precision < 0.5 or recall < 0.5): |
|
|
img_path = os.path.join(source_dir, line_items[0].replace(".txt", ".jpg")) |
|
|
if os.path.exists(img_path): |
|
|
os.system('cp {} {}'.format(img_path, outpt_dir)) |
|
|
sel_list.append((int(id.replace(".txt", "").replace("img", "").replace("_", "")), line)) |
|
|
if id == "ALL": |
|
|
all_eval.write("{} {} {}\n".format( |
|
|
outpt_dir.split('/')[-1], |
|
|
"{}/{}".format(cfg.dis_threshold, cfg.cls_threshold), |
|
|
line)) |
|
|
sel_list = sorted(sel_list, key=lambda its: its[0]) |
|
|
bad_txt.write('\n'.join([its[1] for its in sel_list])) |
|
|
all_eval.close() |
|
|
bad_txt.close() |
|
|
|
|
|
|
|
|
def deal_eval_total_text(debug=False): |
|
|
|
|
|
eval_dir = os.path.join(cfg.output_dir, "Analysis", "output_eval") |
|
|
if not os.path.exists(eval_dir): |
|
|
os.makedirs(eval_dir) |
|
|
|
|
|
print('Computing DetEval in {}/{}'.format(cfg.output_dir, cfg.exp_name)) |
|
|
subprocess.call( |
|
|
['python', 'dataset/total_text/Evaluation_Protocol/Python_scripts/Deteval.py', cfg.exp_name, '--tr', '0.7', |
|
|
'--tp', '0.6']) |
|
|
subprocess.call( |
|
|
['python', 'dataset/total_text/Evaluation_Protocol/Python_scripts/Deteval.py', cfg.exp_name, '--tr', '0.8', |
|
|
'--tp', '0.4']) |
|
|
|
|
|
if debug: |
|
|
source_dir = os.path.join(cfg.vis_dir, '{}_test'.format(cfg.exp_name)) |
|
|
outpt_dir_base = os.path.join(cfg.output_dir, "Analysis", "eval_view", "total_text") |
|
|
if not os.path.exists(outpt_dir_base): |
|
|
mkdirs(outpt_dir_base) |
|
|
|
|
|
outpt_dir1 = os.path.join(outpt_dir_base, "{}_{}_{}_{}_{}" |
|
|
.format(cfg.test_size[0], cfg.test_size[1], cfg.checkepoch, 0.7, 0.6)) |
|
|
osmkdir(outpt_dir1) |
|
|
fid_path1 = '{}/Eval_TotalText_{}_{}.txt'.format(eval_dir, 0.7, 0.6) |
|
|
|
|
|
analysize_result(source_dir, fid_path1, outpt_dir1, "totalText") |
|
|
|
|
|
outpt_dir2 = os.path.join(outpt_dir_base, "{}_{}_{}_{}_{}" |
|
|
.format(cfg.test_size[0], cfg.test_size[1], cfg.checkepoch, 0.8, 0.4)) |
|
|
osmkdir(outpt_dir2) |
|
|
fid_path2 = '{}/Eval_TotalText_{}_{}.txt'.format(eval_dir, 0.8, 0.4) |
|
|
|
|
|
analysize_result(source_dir, fid_path2, outpt_dir2, "totalText") |
|
|
|
|
|
print('End.') |
|
|
|
|
|
|
|
|
def deal_eval_ctw1500(debug=False): |
|
|
|
|
|
eval_dir = os.path.join(cfg.output_dir, "Analysis", "output_eval") |
|
|
if not os.path.exists(eval_dir): |
|
|
os.makedirs(eval_dir) |
|
|
|
|
|
print('Computing DetEval in {}/{}'.format(cfg.output_dir, cfg.exp_name)) |
|
|
subprocess.call(['python', 'dataset/ctw1500/Evaluation_Protocol/ctw1500_eval.py', cfg.exp_name]) |
|
|
|
|
|
if debug: |
|
|
source_dir = os.path.join(cfg.vis_dir, '{}_test'.format(cfg.exp_name)) |
|
|
outpt_dir_base = os.path.join(cfg.output_dir, "Analysis", "eval_view", "ctw1500") |
|
|
if not os.path.exists(outpt_dir_base): |
|
|
mkdirs(outpt_dir_base) |
|
|
|
|
|
outpt_dir = os.path.join(outpt_dir_base, "{}_{}_{}".format(cfg.test_size[0], cfg.test_size[1], cfg.checkepoch)) |
|
|
osmkdir(outpt_dir) |
|
|
fid_path1 = '{}/Eval_ctw1500_{}.txt'.format(eval_dir, 0.5) |
|
|
|
|
|
analysize_result(source_dir, fid_path1, outpt_dir, "ctw1500") |
|
|
|
|
|
print('End.') |
|
|
|
|
|
|
|
|
def deal_eval_icdar15(debug=False): |
|
|
|
|
|
eval_dir = os.path.join(cfg.output_dir, "Analysis", "output_eval") |
|
|
if not os.path.exists(eval_dir): |
|
|
os.makedirs(eval_dir) |
|
|
|
|
|
input_dir = 'output/{}'.format(cfg.exp_name) |
|
|
father_path = os.path.abspath(input_dir) |
|
|
print(father_path) |
|
|
print('Computing DetEval in {}/{}'.format(cfg.output_dir, cfg.exp_name)) |
|
|
subprocess.call(['sh', 'dataset/icdar15/eval.sh', father_path]) |
|
|
|
|
|
if debug: |
|
|
source_dir = os.path.join(cfg.vis_dir, '{}_test'.format(cfg.exp_name)) |
|
|
outpt_dir_base = os.path.join(cfg.output_dir, "Analysis", "eval_view", "icdar15") |
|
|
if not os.path.exists(outpt_dir_base): |
|
|
mkdirs(outpt_dir_base) |
|
|
|
|
|
outpt_dir = os.path.join(outpt_dir_base, "{}_{}_{}".format(cfg.test_size[0], cfg.test_size[1], cfg.checkepoch)) |
|
|
osmkdir(outpt_dir) |
|
|
fid_path1 = '{}/Eval_icdar15.txt'.format(eval_dir) |
|
|
|
|
|
analysize_result(source_dir, fid_path1, outpt_dir, "icdar15") |
|
|
|
|
|
print('End.') |
|
|
|
|
|
pass |
|
|
|
|
|
|
|
|
def deal_eval_TD500(debug=False): |
|
|
|
|
|
eval_dir = os.path.join(cfg.output_dir, "Analysis", "output_eval") |
|
|
if not os.path.exists(eval_dir): |
|
|
os.makedirs(eval_dir) |
|
|
|
|
|
input_dir = 'output/{}'.format(cfg.exp_name) |
|
|
father_path = os.path.abspath(input_dir) |
|
|
print(father_path) |
|
|
print('Computing DetEval in {}/{}'.format(cfg.output_dir, cfg.exp_name)) |
|
|
subprocess.call(['sh', 'dataset/TD500/eval.sh', father_path]) |
|
|
|
|
|
if debug: |
|
|
source_dir = os.path.join(cfg.vis_dir, '{}_test'.format(cfg.exp_name)) |
|
|
outpt_dir_base = os.path.join(cfg.output_dir, "Analysis", "eval_view", "TD500") |
|
|
if not os.path.exists(outpt_dir_base): |
|
|
mkdirs(outpt_dir_base) |
|
|
|
|
|
outpt_dir = os.path.join(outpt_dir_base, "{}_{}_{}".format(cfg.test_size[0], cfg.test_size[1], cfg.checkepoch)) |
|
|
osmkdir(outpt_dir) |
|
|
fid_path1 = '{}/Eval_TD500.txt'.format(eval_dir) |
|
|
|
|
|
analysize_result(source_dir, fid_path1, outpt_dir, "TD500") |
|
|
|
|
|
print('End.') |
|
|
|
|
|
|
|
|
def data_transfer_ICDAR(contours): |
|
|
cnts = list() |
|
|
for cont in contours: |
|
|
rect = cv2.minAreaRect(cont) |
|
|
if min(rect[1][0], rect[1][1]) <= 5: |
|
|
continue |
|
|
points = cv2.boxPoints(rect) |
|
|
points = np.int0(points) |
|
|
|
|
|
|
|
|
cnts.append(points) |
|
|
return cnts |
|
|
|
|
|
|
|
|
def data_transfer_TD500(contours, res_file, img=None): |
|
|
with open(res_file, 'w') as f: |
|
|
for cont in contours: |
|
|
rect = cv2.minAreaRect(cont) |
|
|
if min(rect[1][0], rect[1][1]) <= 5: |
|
|
continue |
|
|
points = cv2.boxPoints(rect) |
|
|
box = np.int0(points) |
|
|
cv2.drawContours(img, [box], 0, (0, 255, 0), 3) |
|
|
|
|
|
cx, cy = rect[0] |
|
|
w_, h_ = rect[1] |
|
|
angle = rect[2] |
|
|
mid_ = 0 |
|
|
if angle > 45: |
|
|
angle = 90 - angle |
|
|
mid_ = w_; |
|
|
w_ = h_; |
|
|
h_ = mid_ |
|
|
elif angle < -45: |
|
|
angle = 90 + angle |
|
|
mid_ = w_; |
|
|
w_ = h_; |
|
|
h_ = mid_ |
|
|
angle = angle / 180 * 3.141592653589 |
|
|
|
|
|
x_min = int(cx - w_ / 2) |
|
|
x_max = int(cx + w_ / 2) |
|
|
y_min = int(cy - h_ / 2) |
|
|
y_max = int(cy + h_ / 2) |
|
|
f.write('{},{},{},{},{}\r\n'.format(x_min, y_min, x_max, y_max, angle)) |
|
|
|
|
|
return img |
|
|
|
|
|
|
|
|
def data_transfer_MLT2017(contours, res_file): |
|
|
with open(res_file, 'w') as f: |
|
|
for cont in contours: |
|
|
rect = cv2.minAreaRect(cont) |
|
|
if min(rect[1][0], rect[1][1]) <= 5: |
|
|
continue |
|
|
ploy_area = cv2.contourArea(cont) |
|
|
rect_area = rect[1][0]*rect[1][1] |
|
|
solidity = ploy_area/rect_area |
|
|
width = rect[1][0] - np.clip(rect[1][0] * (1-np.sqrt(solidity)), 0, 6) |
|
|
height = rect[1][1] - np.clip(rect[1][1] * (1-np.sqrt(solidity)), 0, 4) |
|
|
points = cv2.boxPoints((rect[0], (width, height), rect[2])) |
|
|
points = np.int0(points) |
|
|
p = np.reshape(points, -1) |
|
|
f.write('{},{},{},{},{},{},{},{},{}\r\n' |
|
|
.format(p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 1)) |
|
|
|
|
|
|
|
|
|
|
|
|
