python 图片批量打标签

图片批量打标签

batch_label.py

import os
import cv2
import numpy as np
import shutil

if __name__ == '__main__':
  
    path_in = "D:\Code\image2"    
    path_out = "D:\Code\image3"

    file_in = os.listdir(path_in)       #返回文件夹包含的所有文件名
    num_file_in = len(file_in)          #获取文件数目
    
    #如果，图片名称中有汉字前缀，要先去掉，否则下面img.shape会报错
    #批量图片上写序号
    for i in range(num_file_in):
        
        text = "{}".format(i)
        name = "{}".format(file_in[i])
        img = cv2.imread(os.path.join(path_in, file_in[i]))
    
        #print(img.shape[:2])
        cv2.putText(img,text,(100,img.shape[0]-100),fontFace=cv2.FONT_HERSHEY_TRIPLEX, fontScale=100,color=(0,0,255),thickness=30)
        cv2.imwrite(os.path.join(path_out,name), img)