记录(1)

记录一下

用tensorflow写一个通过CNN模型识别人脸的过程

想要实现的功能:

打开电脑摄像头,可以识别到我的脸并且用绿框框起来,在框上方显示我的名字

原代码来自 教你一步一步用python在图像上做物体检测_python机器学习图片物体检测-CSDN博客

打开摄像头并进行人脸检测代码:

import cv2
# 加载级联分类器模型
faceCascade = cv2.CascadeClassifier("haarcascade_frontalface_default.xml")
# 打开摄像头
video = cv2.VideoCapture(0)
# # 获取图片
# ret, image = video.read()
# # 图像灰度化
# gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
# # 人脸检测
# faces = faceCascade.detectMultiScale(gray, scaleFactor=1.2, minNeighbors=5, minSize=(30, 30),)
# # 用绿框圈出
# for (x, y, width, height) in faces:
#     cv2.rectangle(image, (x, y), (x + width, y + height), (0, 255, 0), 2)
#     # 显示图片
#     cv2.imshow("Face", image)
#     cv2.waitKey(0)

while video.isOpened():
    # 判断视频是否被打开,如果以初始化视频补货,则返回True
    # 读取每一帧的图像
    ret,frame=video.read()
    # 图像灰度化
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    # 人脸检测
    faces = faceCascade.detectMultiScale(gray, scaleFactor=1.2, minNeighbors=5, minSize=(30, 30),)
    # 用绿框圈出
    for (x, y, width, height) in faces:
        cv2.rectangle(frame, (x, y), (x + width, y + height), (0, 255, 0), 2)
        # cv2.putText(, "sunkeying", (x, y - 15))
        cv2.putText(frame, "sunkeying", (x, y - 15), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (0,255,0), 2)

    #retval, frame = cv.VideoCapture.read( [, frame] )
    #retval:为bool类型表示这一帧是否真确读取，正确读取为True，如果文件读取到结尾，它的返回值就为False;
    #frame:表示这一帧的像素点数组
    #如果图像被检测到的话
    if frame is None:
        break
    if ret==True:
        cv2.imshow('windows',frame)
    # 按空格退出
    if ord(' ') == cv2.waitKey(10):
        break

video.release()
cv2.destroyAllWindows()

 对图像进行人脸检测,并裁剪人脸,然后变成100*100大小,保存到文件夹里:

import cv2

faceCascade = cv2.CascadeClassifier("haarcascade_frontalface_default.xml")
image=cv2.imread('mine.jpg')
# 人脸检测
# 图像灰度化
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
# 人脸检测
faces = faceCascade.detectMultiScale(gray, scaleFactor=1.2, minNeighbors=5, minSize=(30, 30),)
print(f"faces:{faces}")
# 用绿框圈出
for (x, y, width, height) in faces:
    # 用绿框圈出
    # cv2.rectangle(resized_image, (x, y), (x + width, y + height), (0, 255, 0), 2)
    # 裁剪图片
    cropped_image = image[y:y + height, x:x + width]

# 原来是173*173,改变尺寸变成100*100
cropped_image2 = cv2.resize(cropped_image, (100, 100))
# 显示图片
cv2.imshow('demo', cropped_image2 )
#保存裁剪后的图片
cv2.imwrite('output_image2.jpg', cropped_image2)

# 等待用户按键后销毁所有窗口
cv2.waitKey(0)
cv2.destroyAllWindows()

很杂乱的代码,还没整理,考完试整理一下,先凑合着看(这个代码是别人的,等回去放链接)

# 将文件夹里的图片进行重命名并且保存到新的文件夹里面
import os
class BatchRename():
    '''
    批量重命名文件夹中的图片文件
    '''

    def __init__(self):
        self.path = 'ABERDEEN_Face'  # 表示需要命名处理的文件夹

    def rename(self):
        filelist = os.listdir(self.path)  # 获取文件路径
        total_num = len(filelist)  # 获取文件长度（个数）
        i = 1  # 表示文件的命名是从1开始的
        for item in filelist:
            if item.endswith('.jpg'):  # 初始的图片的格式为jpg格式的（或者源文件是png格式及其
                # 他格式，后面的转换格式就可以调整为自己需要的格式即可）
                src = os.path.join(os.path.abspath(self.path), item)
                dst = os.path.join(os.path.abspath(self.path), 's' + str(i) + '.jpg')  # 处理后的格式也为jpg格式的，当然这里可以改成png格式
                # dst = os.path.join(os.path.abspath(self.path), '0000' + format(str(i), '0>3s') + '.jpg')    这种情况下的命名格式为0000000.jpg形式，可以自主定义想要的格式
                try:
                    os.rename(src, dst)
                    print('converting %s to %s ...' % (src, dst))
                    i = i + 1
                except:
                    continue
        print('total %d to rename & converted %d jpgs' % (total_num, i))


if __name__ == '__main__':
    demo = BatchRename()
    demo.rename()