update

Author: 刘宇

opencv学习/06检测多个.py

@@ -3,14 +3,17 @@
 
 # 检测函数
 def face_detect_demo(img):
-    # img = cv.resize(img, dsize=(500, 500))
+    print(img.shape)
+    r = 0.5
+    w, h, _ = img.shape
+    img = cv.resize(img, dsize=(int(h * r), int(w * r)))
     # 1 转换为灰度图
     gary = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
     # 加载分类器
     face_detect = cv.CascadeClassifier(
         r"D:\install\opencv\opencv\sources\data\haarcascades\haarcascade_frontalface_alt2.xml"
     )
-    face = face_detect.detectMultiScale(gary, 1.1, 5)
+    face = face_detect.detectMultiScale(gary)
     for x, y, w, h in face:
         cv.rectangle(img, (x, y), (x + w, y + h), color=(0, 0, 255), thickness=2)
     cv.imshow("result", img)

opencv学习/07视频检测.py

@@ -0,0 +1,36 @@
+import cv2 as cv
+
+
+# 检测函数
+def face_detect_demo(img):
+    # 1 转换为灰度图
+    gary = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
+    # 加载分类器
+    face_detect = cv.CascadeClassifier(
+        r"D:\install\opencv\opencv\sources\data\haarcascades\haarcascade_frontalface_alt2.xml"
+    )
+    face = face_detect.detectMultiScale(gary)
+    for x, y, w, h in face:
+        cv.rectangle(img, (x, y), (x + w, y + h), color=(0, 0, 255), thickness=2)
+    cv.imshow("result", img)
+
+
+# 读取摄像头
+cap = cv.VideoCapture(0)
+
+# cap.read()
+
+# face_detect_demo(img)
+# 等待
+while True:
+    flag, frame = cap.read()
+    if not flag:
+        break
+    face_detect_demo(frame)
+    if ord("q") == cv.waitKey(1):  # 这里delay = 1 才不会一直第一帧
+        break
+# 释放内存
+cv.destroyAllWindows()
+
+# 释放摄像头
+cap.release()

opencv学习/08拍照保存信息录入.py

@@ -0,0 +1,18 @@
+import cv2 as cv
+
+cap = cv.VideoCapture(0)
+
+flag = 1
+num = 1
+while (cap.isOpened()):  # 检测是否在开启状态
+    ret_flag, Vshow = cap.read()
+    cv.imshow("Capture", Vshow)
+    k = cv.waitKey(1) & 0xFF  # 按键判断
+    if k == ord("s"):
+        cv.imwrite(f"save/{num}.jpg", Vshow)
+        print("成功 保存 图片", num)
+        num = num + 1
+    elif k == ord(" "):
+        break
+cap.release()
+cv.destroyAllWindows()

opencv学习/09训练数据.py

@@ -0,0 +1,50 @@
+import cv2 as cv
+import os
+from PIL import Image
+import numpy as np
+
+
+def getImageAndLables(path):
+    # 储存人脸数据
+    facesSamples = []
+    # 储存姓名数据
+    ids = []
+    # 储存图片信息
+    imagePaths = [os.path.join(path, f) for f in os.listdir(path)]
+    # 加载分类器
+    face_detector = cv.CascadeClassifier(
+        r"D:\install\opencv\opencv\sources\data\haarcascades\haarcascade_frontalface_alt2.xml"
+    )
+    # 遍历列表中图片
+    for imagePath in imagePaths:
+        # 打开图片，灰度化 PIL有九种不同模式  1 L  P  RGB RGBA  CMYK YCbCr  I F
+        PIL_img = Image.open(imagePath).convert("L")
+
+        # ----------------
+        # ----------------
+        # 将 图像转化为数组  以黑白深浅
+        img_numpy = np.array(PIL_img, "uint8")
+        # 获取图片人脸特征
+        faces = face_detector.detectMultiScale(img_numpy)
+        # 获取每张图片的id和姓名
+        id = int(os.path.split(imagePath)[1].split(".")[0])
+        for x, y, w, h in faces:
+            ids.append(id)
+            facesSamples.append(img_numpy[y: y + h, x:x + w])
+    # 打印脸部特征和id
+    print("id", id)
+    print("fs", facesSamples)
+    return facesSamples, ids
+
+
+if __name__ == '__main__':
+    path = "save/"
+    # 获取图像数组和id标签数值和姓名
+    faces, ids = getImageAndLables(path)
+    # 加载识别器
+
+    recongnizer = cv.face.LBPHFaceRecognizer_create()
+    # 训练
+    recongnizer.train(faces, np.array(ids))
+    # 保存
+    recongnizer.write("trainer/trainer.yml")

opencv学习/10人脸识别.py

@@ -0,0 +1,95 @@
+import cv2
+import numpy as np
+import os
+# coding=utf-8
+import urllib
+import urllib.request
+import hashlib
+
+#加载训练数据集文件
+recogizer=cv2.face.LBPHFaceRecognizer_create()
+recogizer.read('trainer/trainer.yml')
+names=[]
+warningtime = 0
+
+def md5(str):
+    import hashlib
+    m = hashlib.md5()
+    m.update(str.encode("utf8"))
+    return m.hexdigest()
+
+statusStr = {
+    '0': '短信发送成功',
+    '-1': '参数不全',
+    '-2': '服务器空间不支持,请确认支持curl或者fsocket,联系您的空间商解决或者更换空间',
+    '30': '密码错误',
+    '40': '账号不存在',
+    '41': '余额不足',
+    '42': '账户已过期',
+    '43': 'IP地址限制',
+    '50': '内容含有敏感词'
+}
+
+
+def warning():
+    smsapi = "http://api.smsbao.com/"
+    # 短信平台账号
+    user = '13******10'
+    # 短信平台密码
+    password = md5('*******')
+    # 要发送的短信内容
+    content = '【报警】\n原因：检测到未知人员\n地点：xxx'
+    # 要发送短信的手机号码
+    phone = '*******'
+
+    data = urllib.parse.urlencode({'u': user, 'p': password, 'm': phone, 'c': content})
+    send_url = smsapi + 'sms?' + data
+    response = urllib.request.urlopen(send_url)
+    the_page = response.read().decode('utf-8')
+    print(statusStr[the_page])
+
+#准备识别的图片
+def face_detect_demo(img):
+    gray=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)#转换为灰度
+    face_detector=cv2.CascadeClassifier('D:/opencv/opencv/sources/data/haarcascades/haarcascade_frontalface_alt2.xml')
+    face=face_detector.detectMultiScale(gray,1.1,5,cv2.CASCADE_SCALE_IMAGE,(100,100),(300,300))
+    #face=face_detector.detectMultiScale(gray)
+    for x,y,w,h in face:
+        cv2.rectangle(img,(x,y),(x+w,y+h),color=(0,0,255),thickness=2)
+        cv2.circle(img,center=(x+w//2,y+h//2),radius=w//2,color=(0,255,0),thickness=1)
+        # 人脸识别
+        ids, confidence = recogizer.predict(gray[y:y + h, x:x + w])
+        #print('标签id:',ids,'置信评分：', confidence)
+        if confidence > 80:
+            global warningtime
+            warningtime += 1
+            if warningtime > 100:
+               warning()
+               warningtime = 0
+            cv2.putText(img, 'unkonw', (x + 10, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (0, 255, 0), 1)
+        else:
+            cv2.putText(img,str(names[ids-1]), (x + 10, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (0, 255, 0), 1)
+    cv2.imshow('result',img)
+    #print('bug:',ids)
+
+def name():
+    path = './data/jm/'
+    #names = []
+    imagePaths=[os.path.join(path,f) for f in os.listdir(path)]
+    for imagePath in imagePaths:
+       name = str(os.path.split(imagePath)[1].split('.',2)[1])
+       names.append(name)
+
+
+cap=cv2.VideoCapture('1.mp4')
+name()
+while True:
+    flag,frame=cap.read()
+    if not flag:
+        break
+    face_detect_demo(frame)
+    if ord(' ') == cv2.waitKey(10):
+        break
+cv2.destroyAllWindows()
+cap.release()
+#print(names)

opencv学习/11网页视频.py

@@ -0,0 +1,17 @@
+import cv2
+
+
+class CaptureVideo(object):
+    def net_video(self):
+        # 获取网络视频流
+        # cam = cv2.VideoCapture("rtmp://192.168.0.10/live/test")
+        cam = cv2.VideoCapture("rtmp://58.200.131.2:1935/livetv/hunantv")
+        while cam.isOpened():
+            sucess, frame = cam.read()
+            cv2.imshow("Network", frame)
+            cv2.waitKey(1)
+
+
+if __name__ == "__main__":
+    capture_video = CaptureVideo()
+    capture_video.net_video()