用opencv对文件夹下图片通过关键字进行分类存储

用opencv对文件夹下图片通过关键字进行分类存储

参考文章链接：https://blog.csdn.net/weixin_45086637/article/details/92800226?
https://blog.csdn.net/weixin_45204213/article/details/103891914?

需要引入的库

import os
import re
import cv2

需要筛选的关键词 str

str="-1"

正文

def read_directory(director_name):
    for filename in os.listdir(director_name):

        m=re.search(str,filename)
        if m is not None:
            print(filename)
            img = cv2.imread(director_name + "/" + filename)
            cv2.imwrite("E:\data\\rock1" + "/" + filename, img)##存储-1的位置
        else:
            print(filename)
            img = cv2.imread(director_name + "/" + filename)
            cv2.imwrite("E:\data\\rock2" + "/" + filename, img)##存非包含-1的图像位置

read_directory("E:\data\\rock") ##原文件位置

全文

import os
import re
import cv2

str="-1"

def read_directory(director_name):
    for filename in os.listdir(director_name):

        m=re.search(str,filename)
        if m is not None:
            print(filename)
            img = cv2.imread(director_name + "/" + filename)
            cv2.imwrite("E:\data\\rock1" + "/" + filename, img)##存储-1的位置
        else:
            print(filename)
            img = cv2.imread(director_name + "/" + filename)
            cv2.imwrite("E:\data\\rock2" + "/" + filename, img)##存-2的位置

read_directory("E:\data\\rock") ##原文件位置



效果
![原文件](https://img-blog.csdnimg.cn/20210313113201144.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JlYXJzaA==,size_16,color_FFFFFF,t_70#pic_center)

![效果图](https://img-blog.csdnimg.cn/2021031311311668.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JlYXJzaA==,size_16,color_FFFFFF,t_70#pic_center)
![效果图](https://img-blog.csdnimg.cn/20210313113148570.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JlYXJzaA==,size_16,color_FFFFFF,t_70#pic_center)

导入