1. 首先需要将所有的 XML 和 JPEG 文件放在同一个文件夹中,比如命名为 'images'。

  2. 然后创建两个新文件夹,一个命名为 'train',另一个命名为 'test'。

  3. 打开一个文本编辑器,比如记事本,在其中输入以下代码:

import os
import random

# 原始文件夹的路径
path = 'images'

# 训练集的路径
train_path = 'train'

# 测试集的路径
test_path = 'test'

# 划分比例
ratio = 0.8    

# 扫描文件夹中的所有文件
files = os.listdir(path)

# 随机打乱文件列表
random.shuffle(files)

# 计算训练集和测试集的数量
train_num = int(len(files) * ratio)
test_num = len(files) - train_num

# 将文件分成训练集和测试集
for i in range(train_num):
    file = files[i]
    os.rename(os.path.join(path, file), os.path.join(train_path, file))

for i in range(train_num, train_num+test_num):
    file = files[i]
    os.rename(os.path.join(path, file), os.path.join(test_path, file))
  1. 将代码保存为一个 Python 脚本文件,比如 'split_data.py'。

  2. 在命令行中进入到包含脚本文件的文件夹中,然后运行以下命令:

python split_data.py
  1. 执行完命令后,程序会自动将所有的 XML 和 JPEG 文件随机分成训练集和测试集,并分别保存在 'train' 和 'test' 文件夹中。
Fasterrcnn 数据集划分:将 XML 和 JPEG 文件分隔为训练集和测试集

原文地址: https://www.cveoy.top/t/topic/oLna 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录