python怎么根据图片名称分类
时间: 2023-05-18 22:07:22 浏览: 276
CNN实现图片分类,用tensorflow python实现,目前是三类,可以增加类别
5星 · 资源好评率100%
你可以使用Python中的os模块来获取图片名称,然后使用字符串操作函数来分类。具体的代码实现可以参考以下示例:
```python
import os
# 定义图片文件夹路径
img_folder = "/path/to/image/folder"
# 获取图片文件夹中所有的文件名
img_files = os.listdir(img_folder)
# 遍历所有的图片文件名
for img_file in img_files:
# 判断图片文件名是否以特定字符串开头
if img_file.startswith("cat"):
# 如果是猫的图片,则将其移动到猫的文件夹中
os.rename(os.path.join(img_folder, img_file), os.path.join(img_folder, "cat", img_file))
elif img_file.startswith("dog"):
# 如果是狗的图片,则将其移动到狗的文件夹中
os.rename(os.path.join(img_folder, img_file), os.path.join(img_folder, "dog", img_file))
else:
# 如果是其它类型的图片,则将其移动到未分类的文件夹中
os.rename(os.path.join(img_folder, img_file), os.path.join(img_folder, "unclassified", img_file))
```
这段代码可以将图片文件夹中以"cat"开头的图片移动到"cat"文件夹中,以"dog"开头的图片移动到"dog"文件夹中,其它类型的图片移动到"unclassified"文件夹中。
阅读全文