为什么Python Splitfolders不能将数据集拆分为训练和测试文件夹?



我正在尝试使用pythonsplit-folders将一组图像(.bmp格式)数据集拆分为训练和测试文件夹。为此,我使用了下面的代码。代码块成功执行,没有出现任何错误,但没有将图像文件夹拆分为test和train文件夹。我的数据集在"ZZ"文件夹中,我想将数据集分成"Res"文件夹中。我在蟒蛇环境中使用python 3.7.11。我怎样才能解决这个问题?帮助升值。

import os
import splitfolders
input_folder = "zz/"
output = "res" #where you want the split datasets saved. one will be created if none is set
# ratio of split are in order of train/val/test.
#You can change to whatever you want. For train/val sets only, you could do .75, .25 for example.
splitfolders.ratio(input_folder, output, seed=1337, ratio=(.75, .25))

如果您只想训练和测试文件夹第二部分为0,表示验证

splitfolders.ratio(input_folder, output= output_folder, seed=1337, ratio = (0.75, 0, 0.25))

最新更新