图片批量打标签
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)