python图片批量保存（opencv）

python图片批量保存（opencv）

# 保存图片的地址
img_file = "E:/leetcode/Mask_RCNN-master/project/data"
# 确认上述地址是否存在
if not os.path.exists(img_file):
	os.mkdir(img_file)

for i in range(0,2):
    image, dims = draw_pic()
    #此处frame就是你需要保存的图像文件
    frame = draw_shape(image, dims)
    cv2.imwrite(img_file + "/" + '%d.jpg'%i, frame)

不要忘了，引入头文件 cv2 和 os