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)