生成python代码,实现识别花生仁、红豆、绿豆、黄豆、黑豆图片文件,并将文件分类存放
时间: 2023-07-10 07:05:47 浏览: 137
通过 代码 识别 图片
以下是一个可能的 Python 代码实现:
```python
import os
import shutil
from PIL import Image
# 定义类别标签和对应的文件夹名称
class_names = {
"peanut": "peanut_images",
"red_bean": "red_bean_images",
"green_bean": "green_bean_images",
"yellow_bean": "yellow_bean_images",
"black_bean": "black_bean_images"
}
# 创建类别文件夹
for folder_name in class_names.values():
os.makedirs(folder_name, exist_ok=True)
# 遍历所有图片文件
for file_name in os.listdir("image_folder"):
# 加载并缩放图片
img = Image.open(os.path.join("image_folder", file_name)).resize((224, 224))
# 对图片进行分类
if "peanut" in file_name:
class_name = "peanut"
elif "red_bean" in file_name:
class_name = "red_bean"
elif "green_bean" in file_name:
class_name = "green_bean"
elif "yellow_bean" in file_name:
class_name = "yellow_bean"
elif "black_bean" in file_name:
class_name = "black_bean"
else:
continue # 如果不是目标类别的图片,跳过此文件
# 将图片复制到对应的文件夹
shutil.copy(os.path.join("image_folder", file_name), os.path.join(class_names[class_name], file_name))
```
这个程序假设你有一个名为 `image_folder` 的文件夹,其中包含要分类的花生仁、红豆、绿豆、黄豆、黑豆图片文件。对于每个文件,程序会加载并缩放它,然后根据文件名中包含的关键词将其分类到相应的文件夹中。最终,程序将生成五个文件夹,分别包含对应类别的图片文件。
阅读全文