如何从文件夹中选择随机文件

Question

我正试图用以下方式组织我的古典小说数据集

|-- train
     |-- cats
     |-- dogs
|-- validation
     |-- cats
     |-- dogs

我发现一篇文章是这样写的

用于火车文件夹

   search for cat images and move them to the cats folder and do the same for dog images

然后是验证文件夹

  search for image id ranging from 5000–5999 

但对于验证集，我希望从整个狗或猫文件夹中随机获取图像，所以我尝试了这个

for i in range(1000):
    random_file=random.choice(os.listdir(train_cats_dir))
    shutil.move(f'{train_cats_dir}/{random_file}',val_cats_dir)

但是循环大约需要3分钟，并且必须为狗文件夹再次循环。我还尝试将os.listdir（train_cats_dir）放入一个变量中并使用一次，但它选择了一个文件两次

那么，什么是更快的方法呢

我是pytorch的新手，正在尝试使用Imagefolder方法