Python实现随机选取Yolo数据集及标签

随机选取

从数据集中随机取11725张图片做训练集生成新的文件夹，剩余的图片做测试集保存为一个新的文件夹，请用python完成这个功能。

import os
import random
import shutil
# 源数据路径
path = '源数据路径'
# 训练文件夹
train_dir = '训练文件夹'
# 测试文件夹
test_dir = '测试文件夹'
# 列出原始数据集路径下的所有文件
files = os.listdir(path)
# 随机抽取11725张图片
random_files = random.sample(files, 11725)
# 将选出的11725张图片复制到训练文件夹中
for i in random_files:
    shutil.copy(os.path.join(path,i), train_dir)
# 将剩余图片复制到测试文件夹中
for i in files:
    if i not in random_files:
        shutil.copy(os.path.join(path,i), test_dir)

图片与标签文件对应

根据标签文件夹中txt文件名称在图片文件夹中找到对应的jpg图片，并将这些图片生成一个新的文件夹

import os
import shutil
labels_dir = "labels/test"
images_dir = "images/image"
new_dir = "images/test"
if not os.path.exists(new_dir):
    os.makedirs(new_dir)
for label_name in os.listdir(labels_dir):
    label_name = label_name.split('.')[0]
    file_name = label_name + '.jpg'
    file_src = os.path.join(images_dir, file_name)
    file_dst = os.path.join(new_dir, file_name)
    shutil.copyfile(file_src, file_dst)

最后：

如果你想要进一步了解更多的相关知识，可以关注下面公众号联系~会不定期发布相关设计内容包括但不限于如下内容:信号处理、通信仿真、算法设计、matlab appdesigner，gui设计、simulink仿真......希望能帮到你！