视频测试
"""
import argparse
import numpy as np
import cv2
from face_recognizer import FaceRecognizer
parser = argparse.ArgumentParser(description='face_recognization')
parser.add_argument('--face_db_root', type=str, default='data/mask_nomask', help='the root path of target database')
parser.add_argument('--input_video_path', type=str, default='D:/mask_data/02.mp4', help='the path of input video')
parser.add_argument('--output_video_path', type=str, default='output.mp4', help='the path of input video')
args = parser.parse_args()
recognizer = FaceRecognizer()
recognizer.create_known_faces(args.face_db_root)
#recognizer.test_100x()
# 测试视频路径
input_movie = cv2.VideoCapture(args.input_video_path)
# 视频尺寸
video_size = (1024, 720)
fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
# 保存的视频路径
output_movie = cv2.VideoWriter(args.output_video_path, fourcc, 10, video_size)
while True:
# Grab a single frame of video
ret, frame = input_movie.read()
import cv2
import numpy as np
from fastapi import FastAPI, File
from face_recognizer import FaceRecognizer
face_rec = FaceRecognizer()
face_rec.create_known_faces('data/mask_nomask')
app = FastAPI()
@app.get("/")
def root():
return {"message": "Hello World"}
@app.post("/update")
def update():
face_rec.create_known_faces('data/mask_nomask')
return {"result": 'ok'}
@app.post("/file")
def file(file: bytes = File(...)):
image_array = np.frombuffer(file, dtype=np.uint8) # numpy array
img = cv2.imdecode(image_array, cv2.COLOR_RGBA2BGR)
result = face_rec.recognize(img)
return {"result": result}
folder_path = "D:\\Visitor_control_project\\Front_Web\\node_server\\face_recognition\\faces_comp"
#dist_path = "/home/preeth/Downloads"
print("start python")
#c = 0
images = [os.path.join(folder_path, f) for f in os.listdir(folder_path) if os.path.isfile(os.path.join(folder_path, f))]
#parser = argparse.ArgumentParser(description='face_recognization')
#parser.add_argument('--face_db_root', type=str, default='data/mask_nomask', help='the root path of target database')
#parser.add_argument('--input_video_path', type=str, default='D:/mask_data/02.mp4', help='the path of input video')
#parser.add_argument('--output_video_path', type=str, default='output.mp4', help='the path of input video')
#args = parser.parse_args()
recognizer = FaceRecognizer()
recognizer.create_known_faces('./face_recognition/faces_in_server')
#recognizer.test_100x()
#input_movie = cv2.VideoCapture(args.input_video_path)
#video_size = (1024, 720)
#조절 가능성 있음!!!
#video_size = (512, 450)
#fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
#output_movie = cv2.VideoWriter(args.output_video_path, fourcc, 10, video_size)
for i in range(len(images)):
