背景说明:
在做一个瑕疵检测项目的时候,模型需要实时处理摄像头获取的图像进行处理,然后进行OK和NG判别,对于判别的结果要分别保存在不同的文件夹中,同时需要保证图片具有可区分性 。
分析:
考虑使用pathlib根据日期创建文件路径,然后结合图片按日期生成方式保存实现该功能。
#按系统日期生成文件夹,将图片保存在该文件夹中
import cv2
import time
from pathlib import Path
import datetime
img = cv2.imread('test.jpg')
imageName= int(datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S').replace(' ','').replace('-','').replace(':',''))
desc_path = r'C:\Users\Administrator\Desktop\test_33'
# 按日期创建文件夹路径名称
resl_folder = fr"{desc_path}/{time.strftime('%Y%m%d')}"
# 创建文件夹
Path(resl_folder).mkdir(parents=True, exist_ok=True)
cv2.imshow("test", img)
cv2.waitKey(0)
cv2.destroyAllWindows()
# 保存文件
cv2.imwrite(f"{resl_folder}/{imageName}" +'.jpg',img)
print('验证结束!')
结果:
运行结束后,会在对应目录下生成相应的日期文件夹,文件夹中保存有相对应的图片。