python - 如何显示来自狗和猫数据集的随机照片？

Question

我正在尝试使用 Jupyter 笔记本中的 Microsoft 猫狗数据集构建 CNN。现在下面的代码检查前九张狗的照片。如何使它随机选择 9 张狗照片以显示在 3x3 图中？

from matplotlib import pyplot 
from matplotlib.image import imread

folder = '/Users/sujiaying/Downloads/dogs-vs-cats/train/'

for i in range(9):
    pyplot.subplot(330 + 1 + i)
    filename = folder + 'dog.' + str(i) + '.jpg'
    image = imread(filename)
    pyplot.imshow(image)

pyplot.show()

Answer 1

如果你知道有多少张狗图片，你可以使用 numpy！

from numpy.random import default_rng

rng = default_rng()
numbers = rng.choice(numDogs, size=9, replace=False)

然后就用

for i in numbers:

代替

for i in range(9):

如果你不想使用 numpy，那么这样的东西会起作用

numbers = []
while True:
    num = random.randint(0, numDogs)
    if num not in numbers:
        numbers.append(num)
    if len(numbers) == 9:
        break