嗨，我在进行人脸识别的烧瓶服务器上工作，为了将人员添加到数据库中，我制作了一个脚本，如果满足条件，可以从网络摄像头拍摄 30 张照片，这是我的脚本
TOTAL = 0
shape_predictor= "shape_predictor_68_face_landmarks.dat" #dace_landmark
detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor(shape_predictor)
(jStart,jEnd) = face_utils.FACIAL_LANDMARKS_IDXS["jaw"]
(nStart,nEnd)= face_utils.FACIAL_LANDMARKS_IDXS["nose"]
print("[INFO] starting video stream thread...")
vs = VideoStream(src=0).start()
fileStream = False
time.sleep(1.0)
fps= FPS().start()
cv2.namedWindow("test")
while True:
frame = vs.read()
frame = imutils.resize(frame, width=450)
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
rects = detector(gray, 0)
for rect in rects:
shape = predictor(gray, rect)
shape = face_utils.shape_to_np(shape)
jaw = shape[jStart:jEnd]
nose = shape[nStart:nEnd]
l1= math.sqrt((jaw[0][0]-nose[1][0])**2+(jaw[0][1]-nose[1][1])**2)
l2 = math.sqrt((jaw[16][0]-nose[1][0])**2+(jaw[16][1]-nose[1][1])**2)
ratio = l1/l2
m = (jaw[0][1]-jaw[16][1])/(jaw[0][0]-jaw[16][0])
if (ratio <= 1.1 or ratio > .9) & (jaw[0][1]==nose[1][1]) &(m <= .05 or m >-.05 ):
cond = True
else:
cond= False
jawhHull = cv2.convexHull(jaw)
noseHull = cv2.convexHull(nose)
cv2.drawContours(frame, [jawhHull, noseHull], -1, (0, 255, 0), 1)
if cond == True:
TOTAL += 1
frame = vs.read()
time.sleep(.1)
frame2= frame.copy()
img_name = "opencv_frame_{}.png".format(TOTAL)
cv2.imwrite(img_name, frame)
print("{} written!".format(img_name))
cv2.imshow("Frame", frame)
fps.update()
if TOTAL == 30:
break
key2 = cv2.waitKey(1) & 0xFF
if key2 == ord('q'):
break
fps.stop()
cv2.destroyAllWindows()
vs.stop()
我的烧瓶 main.py 看起来像这样
app = Flask(__name__)
@app.route('/face_rec', methods=['POST'])
def face_recognition():
if request.method == 'POST':
# check if the post request has the file part
if 'file' in request.files:
file = request.files.get('file')
dict_res = face_rec(file)
return json.dumps(dict_res)
app.run(host='0.0.0.0', port='5001', debug=True)
face_rec路由是用于面部验证的 API，所以我在这里的问题是我如何不在弹出的 OpenCV 窗口中流式传输网络摄像头，而是在烧瓶路由中流式传输网络摄像头，比如说/注册？
我试过这个教程
https://medium.com/datadriveninvestor/video-streaming-using-flask-and-opencv-c464bf8473d6
但根本不明白，但我不知道这个模板是否是一个很好的起点？
<html>
<head>
<title>Video Streaming Demonstration</title>
</head>
<body>
<h1>Video Streaming Demonstration</h1>
<img id="bg" src="{{ url_for( enrollment') }}">
</body>
</html>
提前感谢您的帮助
完成此操作后，我更改了 main.py，如下所示：
app = Flask(__name__)
@app.route('/')
def index():
return render_template('index.html')
def gen():
TOTAL = 0
shape_predictor= "shape_predictor_68_face_landmarks.dat" #dace_landmark
detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor(shape_predictor)
(jStart,jEnd) = face_utils.FACIAL_LANDMARKS_IDXS["jaw"]
(nStart,nEnd)= face_utils.FACIAL_LANDMARKS_IDXS["nose"]
vs = VideoStream(src=0).start()
fileStream = False
time.sleep(1.0)
fps= FPS().start()
while True:
frame = vs.read()
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
rects = detector(gray, 0)
for rect in rects:
shape = predictor(gray, rect)
shape = face_utils.shape_to_np(shape)
jaw = shape[jStart:jEnd]
nose = shape[nStart:nEnd]
l1= math.sqrt((jaw[0][0]-nose[1][0])**2+(jaw[0][1]-nose[1][1])**2)
l2 = math.sqrt((jaw[16][0]-nose[1][0])**2+(jaw[16][1]-nose[1][1])**2)
ratio = l1/l2
m = (jaw[0][1]-jaw[16][1])/(jaw[0][0]-jaw[16][0])
if (ratio <= 1.1 and ratio > .9) :
cond1 = True
else:
cond1= False
if (jaw[0][1]<= (nose[1][1]+5) and jaw[0][1]> (nose[1][1]-5)):
cond2 = True
else:
cond2= False
if (m <= .05 and m >-.05 ):
cond3 = True
else:
cond3= False
jawhHull = cv2.convexHull(jaw)
noseHull = cv2.convexHull(nose)
if cond1 == True & cond2 == True & cond3 == True:
TOTAL += 1
frame = vs.read()
time.sleep(.01)
frame2= frame.copy()
img_name = "pics/opencv_frame_{}.png".format(TOTAL)
cv2.imwrite(img_name, frame2)
print("{} written!".format(img_name), end="r")
if cond1 == False: cv2.putText(frame, "La nariz no se encuentra centrada", (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2)
if cond3 == False: cv2.putText(frame, "el angulo de la sien no esta alineado", (10, 50), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2)
if cond2 == False: cv2.putText(frame, "quijada no alineada con la nariz", (10, 70), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2)
cv2.drawContours(frame, [mouthHull,left_eyeHull, right_eyehHull, jawhHull, right_eyebrowHull, noseHull, left_eyebrowHull], -1, (0, 255, 0), 1)
cv2.putText(frame, "{} por ciento completado".format(TOTAL), (200, 350), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2)
image = cv2.rectangle(frame, (150,50), (550,400), (150, 150, 50), 2)
fps.update()
if TOTAL == 101:
break
key2 = cv2.waitKey(1) & 0xFF
if key2 == ord('q'):
break
(flag, encodedImage) = cv2.imencode(".jpg", frame)
yield(b'--framern' b'Content-Type: image/jpegrnrn' + bytearray(encodedImage) + b'rn')
fps.stop()
cv2.destroyAllWindows()
vs.stop()
@app.route('/video_feed')
def video_feed():
return Response(gen(), mimetype='multipart/x-mixed-replace; boundary=frame')
if __name__ == '__main__':
# defining server ip address and port
app.run(host='0.0.0.0',port='5000', debug=True)
它只是工作得很好。
如果有人可以帮助我让它在单独的文件中工作，例如 camera.py 和 impor 它在 main.py 我将不胜感激