创建测试和训练数据集时出错

Question

数据集文件包含汽车和卡车的图像。汽车的所有图像都在一个文件中，卡车的所有图像都在另一个文件中。包含汽车图像的文件称为 Car，包含卡车图像的文件称为 truck。我正在尝试创建子文件夹 test 和 train。火车文件应包含数据集中 80% 的洗牌汽车和卡车图像，测试文件应包含数据集中 20% 的洗牌汽车和卡车图像。我尝试过使用 os 方法，但弹出错误。相同的代码和错误如下所示。

代码：

import os 
import shutil
import random
import glob         
os.chdir('C:/Users/akash/Downloads/Datasets')
for c in random.sample(glob.glob('car*'),200):
    shutil.move( c, 'train/car' )
for c in random.sample(glob.glob('truck*'),200):
    shutil.move( c, 'train/truck' )
for c in random.sample(glob.glob('car*'),100):
    shutil.move( c, 'test/car' )
for c in random.sample(glob.glob('truck*'),100):
    shutil.move( c, 'test/car' )
os.chdir('../../')

错误：

ValueError                                Traceback (most recent call last)
<ipython-input-11-9e77a796540b> in <module>
----> 1 for c in random.sample(glob.glob('car*'),200):
      2     shutil.move( c, 'train/car' )
      3 for c in random.sample(glob.glob('truck*'),200):
      4     shutil.move( c, 'train/truck' )
      5 for c in random.sample(glob.glob('car*'),100):

C:\ProgramData\Anaconda3\lib\random.py in sample(self, population, k)
    361         n = len(population)
    362         if not 0 <= k <= n:
--> 363             raise ValueError("Sample larger than population or is negative")
    364         result = [None] * k
    365         setsize = 21        # size of a small set minus size of an empty list

ValueError: Sample larger than population or is negative

Answer 1

改变这个

for c in random.sample(glob.glob('car*'),200):

到这里

population = glob.glob('car*')
for c in random.sample(population,200):

在第二行设置断点，当它命中时，检查population 的值，尤其是长度。它需要 >= 200。

Answer 2

出现错误是因为 glob 模块应在其中查找的目录未以正确的格式指定。确保在当前目录中有目录 'train/car'、'train/truck'、test/car 和 test/truck。把你的代码改成下面的代码sn-p。

代码片段：

import os 
import shutil
import random
import glob         
os.chdir('C:/Users/akash/Downloads/Datasets')

if not os.path.isdir('train/car'):
    os.makedirs('train/car')
if not os.path.isdir('train/truck'):
    os.makedirs('train/truck')

if not os.path.isdir('test/car'):
    os.makedirs('test/car')
if not os.path.isdir('test/truck'):
    os.makedirs('test/truck')

for c in random.sample(glob.glob('./car/*'),200):
    shutil.move( c, 'train/car' )
for c in random.sample(glob.glob('./truck/*'),200):
    shutil.move( c, 'train/truck' )
for c in random.sample(glob.glob('./car/*'),100):
    shutil.move( c, 'test/car' )
for c in random.sample(glob.glob('./truck/*'),100):
    shutil.move( c, 'test/truck' )
os.chdir('../../')