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/hiyB 著作权归作者所有。请勿转载和采集!