python 根据指定名称输出图像

想批量保存图像，根据我规定格式来保存
上一段小代码
我从1开始命名，但加了格式 如001.jpg

#定义全局变量
pic_name  = 1

def save_pic(littlepc):
	#声明全局变量，要在函数save_pic中使用
    global pic_name
    # 数字3是定义为3位数如001，可随意修改需要的
    #pic是当前运行目录下的文件夹的名称，用来保存我批量的图像，请确保已经存在pic文件夹
    #图片保存成jpg格式
    Newdir = os.path.join('pic', str(pic_name).zfill(3) + '.jpg')  
	pic_name +=1
	# Save the cropped image
	#Newdir是拼接好的图像的名称字符串
	cv2.imwrite(Newdir, littlepc)

def write_one_dir_label(data_dir):
	#data_dir原始大图的路径
    for img_file in tqdm(os.listdir(data_dir)):
        if path_is_image(img_file):
        	#文件存在才计算
            img_path = os.path.join(data_dir, img_file)
            save_pic(img_path)  

点赞，收藏！ 谢谢！