将VOC图像数据集按比例随机划分，分割后的图片和标签地址写入新的文本文档

目的：

需要写个脚本将VOC图像数据集按0.7：0.2：0.1比例随机混乱划分，分割后的图片和标签地址写入新的文本文档。

图像文件夹：

标签文件夹：

目标文本文件夹格式

话不多说直接上脚本代码:

import os
import random


def get_files(file_dir):  # 获取所有文件名，并放置到列表
    file_list = []
    for file in os.listdir(file_dir):
        file_list.append(file)
    return file_list

# 图像地址和标签地址
img_dir = './roadsign_voc/images/'
annotation_dir = './roadsign_voc/annotations/'

img_list = get_files(img_dir)
annotation_list = get_files(annotation_dir)

print("Total Sample:" + str(len(img_list)))

# 运行脚本程序时手动输入各个比例，
ratio_train = float(input("输入训练集比例: "))
print(ratio_train)
ratio_val = float(input("输入验证集比例: "))
print