将图片测试集的猫狗分到两个不同的文件夹
时间: 2023-11-22 07:52:10 浏览: 38
可以使用Python的os和shutil模块来实现将图片测试集的猫狗分到两个不同的文件夹的操作。具体的代码如下:
```
import os
import shutil
# 设置测试集图片所在的文件夹路径
test_dir = 'path/to/test/directory'
# 创建存放猫和狗图片的文件夹路径
cat_dir = 'path/to/cat/directory'
dog_dir = 'path/to/dog/directory'
# 创建猫和狗文件夹
os.makedirs(cat_dir, exist_ok=True)
os.makedirs(dog_dir, exist_ok=True)
# 遍历测试集文件夹中的所有图片
for file_name in os.listdir(test_dir):
# 获取图片所属的类别
if 'cat' in file_name:
# 将猫图片复制到猫文件夹中
shutil.copy(os.path.join(test_dir, file_name), cat_dir)
elif 'dog' in file_name:
# 将狗图片复制到狗文件夹中
shutil.copy(os.path.join(test_dir, file_name), dog_dir)
```
以上代码中,首先设置了测试集图片所在的文件夹路径,以及要创建的存放猫和狗图片的文件夹路径。然后,使用os.makedirs函数创建猫和狗文件夹。
接着,遍历测试集文件夹中的所有图片,通过判断文件名中是否包含‘cat’或‘dog’来确定图片所属的类别,然后使用shutil.copy函数将图片复制到对应的文件夹中。最终,就可以将测试集的猫狗分到两个不同的文件夹中。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)