def main():
detector = FaceDetector(model_folder='model',
ctx=mx.cpu(0),
num_worker=4,
accurate_landmark=False)
OriginFace = cv2.imread('face.jpg', cv2.IMREAD_COLOR)
ResultOrigin = detector.detect_face(OriginFace)
P = ResultOrigin[1][0]
origin_points = np.array([(int(P[0]), int(P[5])), (int(P[1]), int(P[6])),
(int(P[2]), int(P[7])), (int(P[3]), int(P[8])),
(int(P[4]), int(P[9]))])
swap.initSwappingModule(OriginFace, origin_points)
camera = cv2.VideoCapture(0)
while True:
grab, frame = camera.read()
img = cv2.resize(frame, (320, 180))
t1 = time.time()
results = detector.detect_face(img)
print('time: ', time.time() - t1)
if results is None:
continue
total_boxes = results[0]
points = results[1]
draw = img.copy()
for b in total_boxes:
cv2.rectangle(draw, (int(b[0]), int(b[1])), (int(b[2]), int(b[3])),
(255, 255, 255))
result = draw.copy()
for p in points:
for i in range(5):
cv2.circle(draw, (p[i], p[i + 5]), 1, (255, 0, 0), 2)
origin_points = np.array([(int(p[0]), int(p[5])),
(int(p[1]), int(p[6])),
(int(p[2]), int(p[7])),
(int(p[3]), int(p[8])),
(int(p[4]), int(p[9]))])
result = swap.swap(result, origin_points)
cv2.imshow("FaceSwap", result)
cv2.imshow("detection result", draw)
if cv2.waitKey(1) & 0xFF == ord('x'):
break
camera.release()
cv2.destroyAllWindows()
def main():
bino_cam = BinoCamera(0, True)
bino_cam.set_width_height((2560, 720))
face_detector = FaceDetector()
# The width of frame is the half of camera width.
distance_predictor = DistancePredictor(90, bino_cam.get_width_height()[0] / 2, bino_cam.get_width_height()[1], 60)
distance = 0
distance_threshold = 2000.0 #mm
monitor_flag = True
while True:
if not bino_cam.capture():
break
Drawer.draw_lines(bino_cam.get_lframe())
Drawer.draw_lines(bino_cam.get_rframe())
left_face = face_detector.detect_face(bino_cam.get_lframe())
right_face = face_detector.detect_face(bino_cam.get_rframe())
if left_face:
Drawer.dotted_rectangle(bino_cam.get_lframe(), (left_face.get_x(), left_face.get_y()),
(left_face.get_x() + left_face.get_width(),
left_face.get_y() + left_face.get_height()), (0, 0, 255), 2)
if right_face:
Drawer.dotted_rectangle(bino_cam.get_rframe(), (right_face.get_x(), right_face.get_y()),
(right_face.get_x() + right_face.get_width(),
right_face.get_y() + right_face.get_height()), (0, 0, 255), 2)
if left_face and right_face:
distance = distance_predictor.predict_distance((left_face.get_x(), left_face.get_y()), (right_face.get_x(), right_face.get_y()))
Drawer.text(bino_cam.get_lframe(), "Distance(m): %.2fm" % (distance / 1000), (20, 40), 1.0, (0, 0, 255), 2)
print(distance / 1000)
if 0 < distance < distance_threshold and monitor_flag:
Controller.set_top_window("Unity 2019.1.8f1 - mainScene.unity - Learn unity editor script - PC, Mac & Linux Standalone <DX11>")
monitor_flag = False
cv2.imshow("I am a good stuff v1.0 @ FPS: %d" % bino_cam.get_fps(), np.hstack((bino_cam.get_lframe(), bino_cam.get_rframe())))
if cv2.waitKey(1) == ord('q'):
break
cv2.destroyAllWindows()
class Drawer:
@staticmethod
def label_face(frame, face_rect: tuple, color=(0, 0, 250)):
x, y, w, h = face_rect
horizon_line_length = int(w / 8)
vertical_line_length = int(h / 8)
center = (int(x + w / 2), int(y + h / 2))
cv2.line(frame, (x - horizon_line_length, center[1]),
(x + horizon_line_length, center[1]), color, 2)
cv2.line(frame, (x + w - horizon_line_length, center[1]),
(x + w + horizon_line_length, center[1]), color, 2)
cv2.line(frame, (center[0], y - vertical_line_length),
(center[0], y + vertical_line_length), color, 2)
cv2.line(frame, (center[0], y + h - vertical_line_length),
(center[0], y + h + vertical_line_length), color, 2)
cv2.circle(frame, center, int(h / 2), color, 2)
if __name__ == '__main__':
img = cv2.imread("assets/timg2.jpg")
face_detector = FaceDetector()
face_detector.detect_face(img)
face_list = face_detector.get_face_list()
if len(face_list) > 0:
for face in face_list:
Drawer.label_face(img, face.get_rect(), color=(0, 0, 200))
cv2.imshow("test Drawer", img)
cv2.waitKey(0)
cv2.destroyAllWindows()
class CameraWindow:
def __init__(self, idx, window):
self.idx = idx
self.update_flag = True
self.info_label = tk.Label(window,
text="Monitor [ %d ] @ %2d (fps)" %
(idx + 1, 0),
font=('Arial', 10))
self.info_label.grid(row=3, column=idx + 1)
img = Image.fromarray(np.full((240, 320), 100))
img_tk = ImageTk.PhotoImage(img)
self.frame_label = tk.Label(window)
self.frame_label.configure(image=img_tk)
self.frame_label.grid(row=4, column=idx + 1, ipadx=5, ipady=10)
self.start_btn = tk.Button(window,
text='Start',
command=self.click_button,
width=39,
font=('Arial', 10))
self.start_btn.grid(row=5, column=idx + 1)
self.split_btn = tk.Button(window,
text='Split This Window',
command=self.split_window,
width=39,
font=('Arial', 10))
self.split_btn.grid(row=6, column=idx + 1)
self.face_detect_flag = tk.BooleanVar()
self.face_detect_checkbutton = tk.Checkbutton(
window,
text="Open Face Detection (Detect face in Split window)",
variable=self.face_detect_flag)
self.face_detect_checkbutton.grid(row=7,
column=idx + 1,
sticky='w',
ipadx=2)
self.face_save_flag = tk.BooleanVar()
self.face_save_checkbutton = tk.Checkbutton(
window,
text="Save Face Image (Save face when face detected)",
variable=self.face_save_flag)
self.face_save_checkbutton.grid(row=8,
column=idx + 1,
sticky='w',
ipadx=2)
self.face_detector = FaceDetector()
self.max_face_store_number = 200
self.thread_list = []
self.tcp = None
def update_frame_label(self):
while self.update_flag:
frame = self.tcp.frame
fps = self.tcp.fps
if frame is not None:
frame = cv2.resize(frame, (320, 240))
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
img = Image.fromarray(frame)
img_tk = ImageTk.PhotoImage(img)
self.frame_label.configure(image=img_tk)
info_text = "Monitor [ %d ] @ %2d (fps)" % (self.idx + 1, fps)
self.info_label.configure(text=info_text)
time.sleep(0.5)
def run_monitor(self):
try:
print("Connecting to the monitor...")
self.tcp = TCPRequest(MONITOR_ADDR[self.idx])
t = threading.Thread(target=self.tcp.start)
print("Connected!")
print("Waiting for Image...")
self.thread_list.append(t)
t.start()
self.update_flag = True
t2 = threading.Thread(target=self.update_frame_label)
self.thread_list.append(t2)
t2.start()
except:
msg.showwarning("Warning", "You must set the monitor before use!")
def stop_monitor(self):
self.update_flag = False
if self.tcp is not None:
print("Stop the monitor...")
self.tcp.stop()
print("Stopped!")
self.tcp = None
else:
msg.showwarning(
"Warning",
"Can't stop monitor, no matched monitor is running.")
def click_button(self):
text = self.start_btn.cget("text")
# Start to Run monitor
if text == "Start":
self.start_btn.configure(text="Stop")
self.run_monitor()
# Stop run monitor
else:
self.start_btn.configure(text="Start")
self.stop_monitor()
def save_face(self, frame, rect):
if not os.path.exists("face_history"):
os.makedirs("face_history")
# remove img dir if this directory is full of pictures
if len(os.listdir("face_history")) > self.max_face_store_number:
os.removedirs("face_history")
os.mkdir("face_history")
x, y, w, h = rect
file_name = os.path.join(
"face_history",
datetime.now().strftime("%Y_%m_%d-%H_%M_%S") + ".jpg")
cv2.imwrite(file_name, frame[y - 10:y + h + 10, x - 10:x + w + 10])
print("Face Image saved as -> face_history/" + file_name)
def show_max_window(self):
while True:
frame = self.tcp.frame.copy()
face_detect_bool = self.face_detect_flag.get()
face_save_bool = self.face_save_flag.get()
# detect the face and label
if face_detect_bool:
self.face_detector.detect_face(frame)
face_list = self.face_detector.get_face_list()
if len(face_list) > 0:
for face in face_list:
Drawer.label_face(frame, face.get_rect())
# save face(s)
if face_save_bool is True:
self.save_face(self.tcp.frame, face.get_rect())
else:
if face_save_bool:
msg.showwarning(
"Warning",
"Open Face Detection before save face image!")
self.face_save_flag.set(False)
# Write Text on the frame
localtime = time.asctime(time.localtime(time.time()))
width = frame.shape[0]
cv2.putText(frame, localtime, (int(0.9 * width), 20),
cv2.FONT_HERSHEY_SIMPLEX, 0.4, (0, 255, 0))
if face_detect_bool is True:
cv2.putText(frame, "Face Detection: True",
(int(0.9 * width), 35), cv2.FONT_HERSHEY_SIMPLEX,
0.4, (0, 255, 0))
else:
cv2.putText(frame, "Face Detection: False",
(int(0.9 * width), 35), cv2.FONT_HERSHEY_SIMPLEX,
0.4, (0, 0, 255))
if face_save_bool is True:
cv2.putText(frame, "Face Save : True",
(int(0.9 * width), 50), cv2.FONT_HERSHEY_SIMPLEX,
0.4, (0, 255, 0))
else:
cv2.putText(frame, "Face Save : False",
(int(0.9 * width), 50), cv2.FONT_HERSHEY_SIMPLEX,
0.4, (0, 0, 255))
cv2.imshow("Split@Monitor %d" % (self.idx + 1), frame)
if cv2.waitKey(1) == ord('q'):
cv2.destroyAllWindows()
break
def split_window(self):
if self.tcp is None:
msg.showwarning("Warning",
"You need to start the monitor at first!")
return
t = threading.Thread(target=self.show_max_window)
self.thread_list.append(t)
t.start()
def join_all_threads(self):
for t in self.thread_list:
t.join()
