# -*- coding: utf-8 -*- # -*- author: jokker -*- import os from JoTools.utils.FileOperationUtil import FileOperationUtil from JoTools.txkjRes.deteRes import DeteRes xml_dir = r"C:\Users\14271\Desktop\连接件训练\xml_angle" save_dir = r"C:\Users\14271\Desktop\连接件训练\xml" img_dir = r"E:\连接件训练数据集" for each_xml_path in FileOperationUtil.re_all_file(xml_dir, endswitch=['.xml']): a = DeteRes(xml_path=each_xml_path) a.img_path = os.path.join(img_dir, os.path.split(each_xml_path)[1][:-4] + '.jpg') a.angle_obj_to_obj() # a.save_to_xml(os.path.join(save_dir, os.path.split(each_xml_path)[1])) a.draw_dete_res( os.path.join(r"C:\Users\14271\Desktop\连接件训练\img", os.path.split(each_xml_path)[1][:-4] + '.jpg'))
json_dir = r"C:\data\004_绝缘子污秽\val\json" a = SegmentJson() dete_res = DeteRes() for each_json_path in list( FileOperationUtil.re_all_file(json_dir, endswitch=['.json']))[20:]: print(each_json_path) a.parse_json_info(each_json_path, parse_img=True, parse_mask=True) dete_res.img = Image.fromarray(a.image_data) for each_obj in a.shapes: print(each_obj.box) box = each_obj.box dete_res.add_obj(box[0], box[1], box[2], box[3], tag=each_obj.label) b = Image.fromarray(a.mask.astype(np.uint8) * 100) b.save(r"C:\Users\14271\Desktop\del\112233.png") dete_res.draw_dete_res(r"C:\Users\14271\Desktop\del\1100.png") # a.save_mask(r"C:\Users\14271\Desktop\del\del.npy") break a.print_as_fzc_format()
loc = res['result']['face_list'][i]['location'] x1, y1 = loc['left'], loc['top'] width, height = loc['width'], loc['height'] x2, y2 = x1 + width, y1 + height face_info.append([int(x1), int(y1), int(x2), int(y2)]) return face_info # OperateDeteRes.crop_imgs(img_dir, xml_dir=img_dir, save_dir=save_dir) # todo 测试正脸的图片 for img_path in FileOperationUtil.re_all_file( img_dir, lambda x: str(x).endswith(('.JPG', '.jpg'))): dete_res = DeteRes(assign_img_path=img_path) res = dete_face(img_path) print(res) for index, each_res in enumerate(res): x1, y1, x2, y2 = each_res dete_res.add_obj(x1=x1, y1=y1, x2=x2, y2=y2, tag='face', assign_id=index) save_path = os.path.join(save_dir, os.path.split(img_path)[1]) dete_res.draw_dete_res(save_path) time.sleep(3)
# -*- coding: utf-8 -*- # -*- author: jokker -*- import time from JoTools.txkjRes.deteRes import DeteRes img_path = r"C:\Users\14271\Desktop\del\test.jpg" xml_path = r"C:\Users\14271\Desktop\del\test.xml" a = DeteRes(xml_path) a.img_path = img_path a.draw_dete_res(r"C:\Users\14271\Desktop\del\draw_res_a.jpg") # # b = a.deep_copy() # # c = DeteRes(json_dict=a.save_to_json()) # # res = c.deep_copy() # # res.print_as_fzc_format() # print('-'*50) # a.print_as_fzc_format() # # print(a.img.width) # print(a.img.height) # # print(res.img.width) # print(res.img.height) #