# -*- coding: utf-8 -*-
# -*- author: jokker -*-
import os
import random
from JoTools.utils.FileOperationUtil import FileOperationUtil
img_dir = r"D:\data\001_fzc_优化相关资料\dataset_fzc\001_train_data_step_1.5\jieya\zd"
save_dir = r"C:\Users\14271\Desktop\train_vit\3"
# for each_img_path in FileOperationUtil.re_all_file(img_dir, endswitch=['.jpg', '.JPG']):
#
# random_num = random.randrange(1, 1000)
#
# print(random_num)
#
# if random_num > 250:
# os.remove(each_img_path)
img_path_list = list(
FileOperationUtil.re_all_file(img_dir, endswitch=['.jpg', '.JPG']))
FileOperationUtil.move_file_to_folder(img_path_list, save_dir, is_clicp=True)
index = 0
for each_xml_path in FileOperationUtil.re_all_file(xml_dir, endswitch=['.xml']):
index += 1
print(index, each_xml_path)
xml_name = os.path.split(each_xml_path)[1]
img_name = xml_name[:-3] + 'jpg'
each_img_path = os.path.join(img_dir, img_name)
if not (os.path.exists(each_xml_path) and os.path.exists(each_img_path)):
print("* 没找到对应的 img 数据： {0}".format(each_img_path))
continue
a = DeteRes(each_xml_path)
# a.filter_by_area(50*100)
if len(a) < 1:
continue
img_path_list.append(each_img_path)
xml_path_list.append(each_xml_path)
FileOperationUtil.move_file_to_folder(img_path_list, img_save_dir, is_clicp=False)
FileOperationUtil.move_file_to_folder(xml_path_list, xml_save_dir, is_clicp=False)
# -*- coding: utf-8 -*- # -*- author: jokker -*- from JoTools.utils.FileOperationUtil import FileOperationUtil assign_dir = r"\\192.168.3.80\大金具-算法\qfm\连接件训练数据集" save_dir = r"C:\Users\14271\Desktop\连接件" FileOperationUtil.move_file_to_folder(FileOperationUtil.re_all_file(assign_dir, endswitch=['.xml']), save_dir, is_clicp=False) config_path = r"D:\Algo\saturn_database\config.ini"
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)