# -*- coding: utf-8 -*-
# -*- author: jokker -*-
import sys
import argparse
from JoTools.operateDeteRes import OperateDeteRes, DeteAcc
from JoTools.txkjRes.deteRes import DeteRes
from JoTools.utils.PrintUtil import PrintUtil
dete_res_standard = r"C:\Users\14271\Desktop\new_gt"
# dete_res_customized = r"C:\Users\14271\Desktop\fzcRust_v1.1.1.0"
dete_res_customized = r"C:\Users\14271\Desktop\fzcRust_v1.0.3.0"
assign_img_path = r""
save_path = r""
a.iou_thershold = 0.3
for each_conf in [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 0.95, 0.99]:
res = a.cal_model_acc(standard_xml_dir=dete_res_standard,
customized_xml_dir=dete_res_customized,
assign_img_dir=assign_img_path,
save_dir=save_path,
save_img=False,
save_xml=True,
assign_conf=each_conf)
res_2 = a.cal_acc_rec(res)
print('-' * 80)
PrintUtil.print(res_2)
from JoTools.utils.PrintUtil import PrintUtil
def parse_args():
"""Parse input arguments."""
parser = argparse.ArgumentParser(description='run model')
parser.add_argument('--img_dir', dest='img_dir',type=str)
parser.add_argument('--endswitch', dest='endswitch',type=str, default=None)
assign_args = parser.parse_args()
return assign_args
if __name__ == "__main__":
if len(sys.argv) > 1:
args = parse_args()
PrintUtil.print(args)
if not args.endswitch is None:
endswitch = args.endswitch.split(',')
else:
endswitch = None
OperateDeteRes.count_assign_dir(args.img_dir, endswitc=endswitch)
else:
img_dir = r"D:\data\001_fzc_优化相关资料\dataset_fzc\001_train_data_step_1.5\解压"
# OperateDeteRes.count_assign_dir(img_dir, endswitc=['.xml', '.jpg'])
OperateDeteRes.count_assign_dir(img_dir, endswitc=['.jpg', '.JPG', '.png', '.PNG'])
each_xml_path = os.path.join(
new_xml_dir,
FileOperationUtil.bang_path(each_img_path)[1] + '.xml')
#
each_new_img_path = os.path.join(region_img_dir,
os.path.split(each_img_path)[1])
each_new_xml_path = os.path.join(region_xml_dir,
os.path.split(each_xml_path)[1])
if os.path.exists(each_new_img_path):
print("* img path exists : {0}".format(each_new_img_path))
continue
if os.path.exists(each_new_xml_path):
print("* xml path exists : {0}".format(each_new_xml_path))
continue
new_img_path_list.append(each_img_path)
new_xml_path_list.append(each_xml_path)
PrintUtil.print(new_img_path_list)
print('-' * 30)
PrintUtil.print(new_xml_path_list)
FileOperationUtil.move_file_to_folder(new_img_path_list,
region_img_dir,
is_clicp=True)
FileOperationUtil.move_file_to_folder(new_xml_path_list,
region_xml_dir,
is_clicp=True)