图片标注并保存

import os
import sys
import xml.etree.ElementTree as ET
import glob
import cv2
def save_image(image,addr,num):
    address = addr + str(num)+ '.jpg'
    cv2.imwrite(address,image)

def pic_label(indir,outdir):
    i=0
    seq_dir = os.path.expanduser(indir)
    img_files=glob.glob(seq_dir + '*.jpg')
    img_files.sort(key = lambda x: int(x[19 :-4]))
    for file in img_files:
        i=i+1
        img = cv2.imread(file)
        cv2.rectangle(img, (114, 150), (412, 333), (0, 255, 0), thickness=2)  # 生成原始值
        save_image(img,'./try/image',i)

indir='./008CXM 9-40/'   #pic目录
outdir='./try/'  #保存目录

pic_label(indir,outdir)