python按指定数量划分图片与对应的xml数据(.png格式为列)
import os
import shutil
input_dir = r"E:\data-to-gaof\input-data" # 输入图片所在路径
output_dir = r"E:\data-to-gaof\output-data" # 指定输出位置
data_num = 400 # 每个文件夹预计存放多少个数据png_list = []
for root,dir,data in os.walk(input_dir):for fname in [x for x in data if x.endswith(".png")]:png_path = os.path.join(input_dir,fname)png_list.append(png_path)# 确定划分为多少个文件夹
all_list_len = len(png_list)/data_num
if all_list_len > int(all_list_len):all_list_len = int(all_list_len) + 1for j in range(int(all_list_len)):if len(png_list) > 0:print("\n",j,"--------------------------------------- the all data num:",len(png_list))if j != (int(all_list_len) - 1):for i,png in enumerate(png_list):xml = png.replace(".png",".xml")if i < data_num:save_dir = os.path.join(output_dir, str(j))if os.path.isdir(save_dir) == False:os.mkdir(save_dir)shutil.move(png, save_dir)try:shutil.move(xml, save_dir)del png_list[i]except:del png_list[i]else:breakelse:for png in png_list:xml = png.replace(".png", ".xml")save_dir = os.path.join(output_dir, str(j))if os.path.isdir(save_dir) == False:os.mkdir(save_dir)shutil.move(png, save_dir)try:shutil.move(xml, save_dir)except:continueelse:print("data list is None or No find data!")
python按指定数量划分图片与对应的xml数据(.png格式为列)
import os
import shutil
input_dir = r"E:\data-to-gaof\input-data" # 输入图片所在路径
output_dir = r"E:\data-to-gaof\output-data" # 指定输出位置
data_num = 400 # 每个文件夹预计存放多少个数据png_list = []
for root,dir,data in os.walk(input_dir):for fname in [x for x in data if x.endswith(".png")]:png_path = os.path.join(input_dir,fname)png_list.append(png_path)# 确定划分为多少个文件夹
all_list_len = len(png_list)/data_num
if all_list_len > int(all_list_len):all_list_len = int(all_list_len) + 1for j in range(int(all_list_len)):if len(png_list) > 0:print("\n",j,"--------------------------------------- the all data num:",len(png_list))if j != (int(all_list_len) - 1):for i,png in enumerate(png_list):xml = png.replace(".png",".xml")if i < data_num:save_dir = os.path.join(output_dir, str(j))if os.path.isdir(save_dir) == False:os.mkdir(save_dir)shutil.move(png, save_dir)try:shutil.move(xml, save_dir)del png_list[i]except:del png_list[i]else:breakelse:for png in png_list:xml = png.replace(".png", ".xml")save_dir = os.path.join(output_dir, str(j))if os.path.isdir(save_dir) == False:os.mkdir(save_dir)shutil.move(png, save_dir)try:shutil.move(xml, save_dir)except:continueelse:print("data list is None or No find data!")