效果如上图所示
import cv2
import numpy as np
from flask import Flask, render_template, Response
app = Flask(__name__)
video_writer = None
class Camera:
_instance = None
def __new__(cls):
if cls._instance is None:
cls._instance = super().__new__(cls)
cls._instance.camera = cv2.VideoCapture(0)
return cls._instance
def __del__(self):
self.camera.release()
def get_frame(self):
success, frame = self.camera.read()
if not success:
return None
else:
ret, buffer = cv2.imencode('.jpg', frame)
frame = buffer.tobytes()
return frame
camera = Camera()
def convert_bytes_to_frame(frame_bytes):
# 将bytes类型的图像数据解码为OpenCV图像格式
frame_array = np.frombuffer(frame_bytes, dtype=np.uint8)
frame = cv2.imdecode(frame_array, flags=cv2.IMREAD_COLOR)
return frame
# 获取视频流
def get_video_stream():
global camera,video_writer
while True:
if camera is None:
break
frame = camera.get_frame()
if frame is None:
break
else:
frameT=convert_bytes_to_frame(frame)
# frameT = cv2.cvtColor(convert_bytes_to_frame(frame), cv2.COLOR_BGR2RGB) # 转换颜色空间
gray = cv2.cvtColor(frameT, cv2.COLOR_RGB2GRAY) # 转换为灰度图像
# 加载人脸检测器
face = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
# 加载眼睛检测器
eye = cv2.CascadeClassifier('haarcascade_eye.xml')
faces = face.detectMultiScale(gray) # 执行人脸检测
for x, y, w, h in faces:
cv2.rectangle(frameT, (x, y), (x + w, y + h), (255, 0, 0), 2) # 绘制矩形标注人脸
roi_eye = gray[y:y + h, x:x + w] # 根据人脸获得眼睛的检测范围
eyes = eye.detectMultiScale(roi_eye) # 在人脸范围内检测眼睛
for (ex, ey, ew, eh) in eyes: # 标注眼睛
cv2.circle(frameT[y:y + h, x:x + w], (int(ex + ew / 2),
int(ey + eh / 2)), int(max(ew, eh) / 2), (0, 255, 0), 2)
if video_writer is not None:
video_writer.write(convert_bytes_to_frame(frame))
ret, buffer = cv2.imencode('.jpg', frameT)
frame = buffer.tobytes()
yield (b'--frame\r\n'
b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
# 关闭摄像头
def close_camera():
global camera
camera = None
@app.route('/open_camera', methods=['POST'])
def open_camera_route():
global camera
camera = Camera()
return Response(get_video_stream(), mimetype='multipart/x-mixed-replace; boundary=frame')
@app.route('/close_camera', methods=['POST'])
def close_camera_route():
close_camera()
return 'Camera closed'
@app.route('/')
def index():
return render_template('index3.html')
@app.route('/video_feed')
def video_feed():
return Response(get_video_stream(), mimetype='multipart/x-mixed-replace; boundary=frame')
@app.route('/start_recording')
def start_recording():
global video_writer
video_writer = cv2.VideoWriter('output.avi', cv2.VideoWriter_fourcc(*'XVID'), 20.0, (640, 480))
return 'Recording started'
@app.route('/stop_recording')
def stop_recording():
global video_writer
video_writer.release()
video_writer = None
return 'Recording stopped'
if __name__ == '__main__':
app.run(debug=True)
完整代码下载地址:https://download.csdn.net/download/douyu0814/88475010
联系客服