facenet, mask_detector
스파르타 강의 3주차
import cv2
import numpy as np
from tensorflow.keras.applications.mobilenet_v2 import preprocess_input
from tensorflow.keras.models import load_model
facenet = cv2.dnn.readNet('models/deploy.prototxt',
"models/res10_300x300_ssd_iter_140000.caffemodel")
model = load_model('models/mask_detector.model')
cap = cv2.VideoCapture(0)
while True:
ret, img = cap.read()
if not ret:
break
h, w, c = img.shape
blob = cv2.dnn.blobFromImage(img, size=(300, 300), mean=(104., 177., 123.))
facenet.setInput(blob)
dets = facenet.forward()
for i in range(dets.shape[2]):
confidence = dets[0, 0, i, 2]
if confidence < 0.5:
continue
x1 = int(dets[0, 0, i, 3] * w)
y1 = int(dets[0, 0, i, 4] * h)
x2 = int(dets[0, 0, i, 5] * w)
y2 = int(dets[0, 0, i, 6] * h)
face = img[y1:y2, x1:x2]
face_input = cv2.resize(face, dsize=(224, 224))
face_input = cv2.cvtColor(face_input, cv2.COLOR_BGR2RGB)
face_input = preprocess_input(face_input)
face_input = np.expand_dims(face_input, axis=0)
mask, nomask = model.predict(face_input).squeeze()
if mask > nomask:
color = (0, 255, 0)
else:
color = (0, 0, 255)
cv2.rectangle(img, pt1=(x1, y1), pt2=(x2, y2), color=color, thickness=2)
cv2.putText(img, text=str(confidence), org=(x1, y1), fontFace=cv2.FONT_HERSHEY_DUPLEX,
fontScale=1, color=(0, 0, 255), thickness=2)
img = cv2.resize(img, dsize=(640, 360))
cv2.imshow('cam', img)
if cv2.waitKey(1) == ord("q"):
break
Author And Source
이 문제에 관하여(facenet, mask_detector), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@thissa/facenet-maskdetector저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)