python 将图片按命名的类别保存到相应的文件夹中

将图片根据命名，分成7类，并保存到7个文件夹

import cv2import osi="0"base_path="data/image/jaffe_48/"imglist=os.listdir(base_path)for img_path in imglist: #print(img_path) #print(img_path.find("AN")+1) if (img_path.find("AN")+1): i="0" elif (img_path.find("DI")+1): i = "1" elif (img_path.find("FE") + 1): i = "2" elif (img_path.find("HA")+1): i = "3" elif (img_path.find("SA") + 1): i = "4" elif (img_path.find("SU") + 1): i = "5" elif (img_path.find("NE") + 1): i = "6" if not os.path.exists(os.path.join(base_path,i)): os.makedirs(os.path.join(base_path,i)) img=cv2.imread(os.path.join(base_path,img_path)) f="{}/{}".format("data/image/jaffe_48",i) cv2.imwrite("{}/{}.jpg".format(f,img_path.split(".jpg")[0]),img)

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。