Fasterrcnn 数据集划分:将 XML 和 JPEG 文件分隔为训练集和测试集
-
首先需要将所有的 XML 和 JPEG 文件放在同一个文件夹中,比如命名为 'images'。
-
然后创建两个新文件夹,一个命名为 'train',另一个命名为 'test'。
-
打开一个文本编辑器,比如记事本,在其中输入以下代码:
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))
-
将代码保存为一个 Python 脚本文件,比如 'split_data.py'。
-
在命令行中进入到包含脚本文件的文件夹中,然后运行以下命令:
python split_data.py
- 执行完命令后,程序会自动将所有的 XML 和 JPEG 文件随机分成训练集和测试集,并分别保存在 'train' 和 'test' 文件夹中。
原文地址: https://www.cveoy.top/t/topic/oLna 著作权归作者所有。请勿转载和采集!