最新消息: USBMI致力于为网友们分享Windows、安卓、IOS等主流手机系统相关的资讯以及评测、同时提供相关教程、应用、软件下载等服务。

python按指定数量划分图片与对应的xml数据(.png格式为列)

IT圈 admin 7浏览 0评论

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!")

发布评论

评论列表 (0)

  1. 暂无评论