最近在做人臉識別，需要對人臉數(shù)據(jù)集進(jìn)行處理，對一張或批量圖像的人臉進(jìn)行切割分離，并且另保存下來。受網(wǎng)上開源工具的啟發(fā)，在借鑒他人的基礎(chǔ)上進(jìn)行了改進(jìn)，使得更加方便實(shí)用。

以下代碼是改進(jìn)版，分為兩部分功能：

一張人臉圖片切割并顯示，不保存

一張/批量人臉圖像切割并保存

在這里需要申明一下，尊重原著！（轉(zhuǎn)發(fā)需要標(biāo)注一下原著信息）

原創(chuàng)版：

? Author: ? ?coneypo
? Blog: ? ? ? http://www.cnblogs.com/AdaminXie
? GitHub: ? https://github.com/coneypo/Dlib_face_cut

改進(jìn)版：

Improver: ?Cai_90hou

Blog: ? ? ? ??https://blog.csdn.net/qq_38677310/article/details/84702662

Github: ? ? ?

#crop_faces_show.pyimport dlib # 人臉識別的庫dlib import numpy as np # 數(shù)據(jù)處理的庫numpy import cv2 # 圖像處理的庫OpenCv# Dlib 正向人臉檢測器 detector = dlib.get_frontal_face_detector()# 讀取圖像 path = "faces_for_test/"img = cv2.imread(path+"test_faces_1.jpg")# Dlib 檢測 dets = detector(img, 1)print("檢測到的人臉數(shù) / faces ：", len(dets), "\n")# 記錄人臉矩陣大小 height_max = 0 width_sum = 0# 計(jì)算要生成的圖像 img_blank 大小 for k, d in enumerate(dets):# 計(jì)算矩形大小# (x,y), (寬度width, 高度height)pos_start = tuple([d.left(), d.top()])pos_end = tuple([d.right(), d.bottom()])# 計(jì)算矩形框大小height = d.bottom()-d.top()width = d.right()-d.left()# 處理寬度width_sum += width# 處理高度if height > height_max:height_max = heightelse:height_max = height_max# 繪制用來顯示人臉的圖像的大小 print("窗口大小：", '\n', "高度 / height :", height_max, '\n', "寬度 / width : ", width_sum)# 生成用來顯示的圖像 img_blank = np.zeros((height_max, width_sum, 3), np.uint8)# 記錄每次開始寫入人臉像素的寬度位置 blank_start = 0# 將人臉填充到img_blank for k, d in enumerate(dets):height = d.bottom()-d.top()width = d.right()-d.left()# 填充for i in range(height):for j in range(width):img_blank[i][blank_start+j] = img[d.top()+i][d.left()+j]# 調(diào)整圖像blank_start += widthcv2.namedWindow("img_faces", 0) cv2.imshow("img_faces", img_blank) cv2.waitKey(0) #crop_faces_save.pyimport dlib # 人臉識別的庫dlib import numpy as np # 數(shù)據(jù)處理的庫numpy import cv2 # 圖像處理的庫OpenCv import os#切割后的人臉序號，為全局變量 image_num = 0# Dlib 正向人臉檢測器 detector = dlib.get_frontal_face_detector()# 讀取圖像的路徑 path_read = "faces_for_test/" #img = cv2.imread(path_read+"test_faces_3.jpg")# 用來存儲生成的單張人臉的路徑 path_save = "faces_separated/"#將文件夾中待處理的圖片存于列表中 imgs_read = os.listdir(path_read) #源地址文件夾 imgs_write = os.listdir(path_save) #存儲地址文件夾# Delete old images def clear_images():for img in imgs_write:os.remove(path_save + img)print("clean finish", '\n')#處理一張圖片： def cut_one_photo(img):global image_num# Dlib 檢測faces = detector(img, 1)print("人臉數(shù)：", len(faces))for k, d in enumerate(faces):# 計(jì)算矩形大小# (x,y), (寬度width, 高度height)pos_start = tuple([d.left(), d.top()])pos_end = tuple([d.right(), d.bottom()])# 計(jì)算矩形框大小height = d.bottom()-d.top()width = d.right()-d.left()# 根據(jù)人臉大小生成空的圖像img_blank = np.zeros((height, width, 3), np.uint8)#復(fù)制人臉for i in range(height):for j in range(width):img_blank[i][j] = img[d.top()+i][d.left()+j]# cv2.imshow("face_"+str(k+1), img_blank)# 保存在本地#print("Save to:", path_save+"CWH"+str(image_num+1)+".jpg")print("Save to:", "img_face" + str(image_num + 1) + ".jpg")cv2.imwrite(path_save + "img_face" + str(image_num + 1) + ".jpg", img_blank)image_num += 1#批量處理圖片 def cut_batch_photoes():clear_images() #清除原有圖片for img in imgs_read:try:path = os.path.join(path_read, img)image = cv2.imread(path)cut_one_photo(image)print('\n')except:continueprint('Successful operation!\n')print('total number of faces: ', image_num)def main():# 批量處理圖片并保存cut_batch_photoes()if __name__ == '__main__':main()

?

總結(jié)

以上是生活随笔為你收集整理的人脸图像切割分离工具的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。