将某一类下面的具体类别数据集整合在一起，并按照7:3分为训练集、测试集

最新推荐文章于 2024-04-16 15:26:37 发布

ShaoDu

最新推荐文章于 2024-04-16 15:26:37 发布

阅读量4.4k

点赞数 1

分类专栏：深度学习与机器学习文章标签：数据集整合数据集分为训练集、测试集

本文链接：https://blog.csdn.net/ShaoDu/article/details/90901704

版权

博客内容讲述了如何将200类鸟的图片数据集进行整理，统一转移到一个目录下，并按照VOC数据集的命名规范进行命名。接着，数据集被划分为70%的训练集和30%的测试集，每部分的图片名被记录到相应的txt文件中。

摘要由CSDN通过智能技术生成

1、首先将类别目录下的具体类别的数据集转移到一个大部目录下
这是鸟的数据集，下面分成200类。将这200类下的图片全部按照voc数据集命名格式转移到bird_image文件目录下：

from PIL import Image
import glob
url = '/home/lf/桌面/bird_image/images/'#图片存储的文件夹名称
n=11788#图片的数量
array = np.arange(n)#产生长度为n的序列
np.random.shuffle(array)#将arrray序列随机排列
#把path文件夹下以及其子文件下的所有.jpg图片移动到new_path文件夹下
def moveImg(path,new_path):
    img=Image.open(path)
    img.save(os.path.join(new_path,os.path.basename(path)))
bird_path="/home/lf/桌面/bird_image/bird_image"
listName=os.listdir(url)
#转移数据集
for i in range(len(listName)):
    name=listName[i]
    image_files=url+name
    for image_name in glob.glob(image_files+"/*.jpg"):
        moveImg(image_name,bird_path)    
bird_list=os.listdir(bird_path)
#将转移的数据集规范命名格式
for i in range(len(bird_list)):
    if bird_list[i].endswith(&#