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'])
# -*- coding: utf-8 -*- # -*- author: jokker -*- import random import os import shutil import cv2 import PIL.Image as Image from JoTools.operateDeteRes import OperateDeteRes from JoTools.txkjRes.deteRes import DeteRes, DeteObj from JoTools.utils.FileOperationUtil import FileOperationUtil from JoTools.utils.RandomUtil import RandomUtil xml_dir = r"/home/ldq/tj_dete/merge" save_dir = r"/home/ldq/tj_dete/merge_new" OperateDeteRes.get_class_count(xml_dir, print_count=True) for each_xml_path in FileOperationUtil.re_all_file(xml_dir, endswitch=['.xml']): # a = DeteRes(each_xml_path) a.filter_by_tags(need_tag=["2"]) save_path = os.path.join(save_dir, os.path.split(each_xml_path)[1]) if len(a) > 0: a.save_to_xml(save_path) OperateDeteRes.get_class_count(save_dir, print_count=True)
parser.add_argument('--split_by_tag', dest='split_by_tag',type=str, default=True) parser.add_argument('--augment_parameter', dest='augment_parameter',type=str, default="0,0,0,0") parser.add_argument('--include_tag_list', dest='include_tag_list',type=str, default=None) parser.add_argument('--exclude_tag_list', dest='exclude_tag_list',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) # 执行 OperateDeteRes.get_xml_from_crop_img(region_img_dir=args.img_dir, crop_dir=args.crop_dir, save_xml_dir=args.save_dir) # 打印结果 OperateDeteRes.get_class_count(args.save_dir, print_count=True) else: img_dir = r"C:\Users\14271\Desktop\fzcRust_data" crop_dir = r"F:\20211019_防震锤锈蚀数据清洗\crop" save_dir = r"F:\20211019_防震锤锈蚀数据清洗\xml" OperateDeteRes.get_xml_from_crop_img(region_img_dir=img_dir, crop_dir=crop_dir, save_xml_dir=save_dir) OperateDeteRes.get_class_count(save_dir, print_count=True)
if args.exclude_tag_list: exclude_tag_list = args.exclude_tag_list.split(',') else: exclude_tag_list = None if args.include_tag_list: include_tag_list = args.include_tag_list.split(',') else: include_tag_list = None # 裁剪 OperateDeteRes.crop_imgs(args.img_dir, args.xml_dir, args.save_dir, split_by_tag=args.split_by_tag, augment_parameter=augment_parameter, exclude_tag_list=exclude_tag_list, include_tag_list=include_tag_list) else: img_dir = r"D:\data\001_fzc_优化相关资料\dataset_fzc\000_0_标准测试集\JPEGImages" # xml_dir = r"C:\Users\14271\Desktop\fangtian_test\save_res" xml_dir = r"C:\Users\14271\Desktop\xml_tmp" save_dir = r"C:\Users\14271\Desktop\crop" OperateDeteRes.crop_imgs(img_dir, xml_dir, save_dir, split_by_tag=True, augment_parameter=[0.3, 0.3, 0.3, 0.3]) # OperateDeteRes.crop_imgs(img_dir, xml_dir, save_dir, split_by_tag=True, augment_parameter=[0.15, 0.15, 0.15, 0.15], save_augment=True, include_tag_list=["bljyz", "tcjyz"])
# -*- coding: utf-8 -*- # -*- author: jokker -*- import sys import argparse from JoTools.operateDeteRes import OperateDeteRes from JoTools.utils.PrintUtil import PrintUtil def parse_args(): """Parse input arguments.""" parser = argparse.ArgumentParser(description='run model') parser.add_argument('--xml_dir', dest='xml_dir', type=str) assign_args = parser.parse_args() return assign_args if __name__ == "__main__": if len(sys.argv) > 1: args = parse_args() PrintUtil.print(args) OperateDeteRes.get_class_count(args.xml_dir, print_count=True) else: # xml_dir = r"C:\Users\14271\Desktop\save_res" # xml_dir = r"C:\Users\14271\Desktop\docker-0.0.1\save_res" xml_dir = r"C:\Users\14271\Desktop\结果对比\merge" OperateDeteRes.get_class_count(xml_dir, print_count=True)
# -*- coding: utf-8 -*- # -*- author: jokker -*- import os from JoTools.txkjRes.deteRes import DeteRes from JoTools.txkj.imageAugmentation import ImageAugmentation from JoTools.utils.FileOperationUtil import FileOperationUtil from JoTools.operateDeteRes import OperateDeteRes, OperateTrainData # img_dir = r"C:\data\fzc_优化相关资料\dataset_fzc\000_train_data_step_1\JPEGImages" # xml_dir = r"C:\data\fzc_优化相关资料\dataset_fzc\000_train_data_step_1\Annotations_broken" # save_dir = r"C:\data\fzc_优化相关资料\dataset_fzc\000_train_data_step_1\crop_with_xml" # index = 0 # for each_xml_path in FileOperationUtil.re_all_file(xml_dir, endswitch=['.xml']): # index += 1 # print(index, each_xml_path) # each_img_path = os.path.join(img_dir, FileOperationUtil.bang_path(each_xml_path)[1] + '.jpg') # a = DeteRes(each_xml_path, assign_img_path=each_img_path) # a.crop_with_xml(augment_parameter=[0.5,0.5,0.5,0.5], save_dir=save_dir, split_by_tag=True) xml_dir = r"C:\data\fzc_优化相关资料\dataset_fzc\000_train_data_step_1\fix_obj_range\crop_xml" img_dir = r"C:\data\fzc_优化相关资料\dataset_fzc\000_train_data_step_1\JPEGImages" save_dir = r"C:\data\fzc_优化相关资料\dataset_fzc\000_train_data_step_1\fix_obj_range\xml" OperateDeteRes.get_xml_from_crop_xml(xml_dir=xml_dir, region_img_dir=img_dir, save_xml_dir=save_dir)
augment_parameter = None if args.exclude_tag_list: exclude_tag_list = args.exclude_tag_list.split(',') else: exclude_tag_list = None if args.include_tag_list: include_tag_list = args.include_tag_list.split(',') else: include_tag_list = None # 裁剪 OperateDeteRes.crop_imgs_angles(args.img_dir, args.xml_dir, args.save_dir, split_by_tag=args.split_by_tag, augment_parameter=augment_parameter, exclude_tag_list=exclude_tag_list) else: img_dir = r"C:\Users\14271\Desktop\new_data\data" xml_dir = r"C:\Users\14271\Desktop\new_data\data" save_dir = r"C:\Users\14271\Desktop\new_data\crop" # OperateDeteRes.crop_imgs(img_dir, xml_dir, save_dir, split_by_tag=True, augment_parameter=[0.1, 0.1, 0.1, 0.1]) # OperateDeteRes.crop_imgs(img_dir, xml_dir, save_dir, split_by_tag=False) OperateDeteRes.crop_imgs_angles(img_dir, xml_dir, save_dir, split_by_tag=False)
# -*- coding: utf-8 -*- # -*- author: jokker -*- from PIL import Image from JoTools.operateDeteRes import OperateDeteRes from JoTools.txkjRes.deteRes import DeteRes, DeteObj, DeteAngleObj from JoTools.utils.FileOperationUtil import FileOperationUtil img_dir = r"C:\Users\14271\Desktop\结果对比\img" # xml_dir = r"C:\Users\14271\Desktop\结果对比\standard_oly_nc" xml_dir = r"C:\Users\14271\Desktop\结果对比\merge_update_only_nc" res_dir = r"C:\Users\14271\Desktop\结果对比\draw" OperateDeteRes.draw_tags(img_dir=img_dir, xml_dir=xml_dir, save_dir=res_dir)