def make_face_boxed_image(img, feat, color=utils.RED, thickness=4, alpha=0.5, show_conf_=True): if feat.face_pos_arr is None or feat.face_conf_arr is None: return img thickness = thickness if show_conf_ else -1 for pos, conf in zip(feat.face_pos_arr, feat.face_conf_arr): img = utils.draw_box_on_img(img, pos, color=color, thickness=thickness, alpha=alpha) if show_conf_: img = cv2.putText(img, "{:3d}".format(conf), (pos[0] + 4, pos[3] - 4), cv2.FONT_HERSHEY_SIMPLEX, 1, utils.WHITE, 5) img = cv2.putText(img, "{:3d}".format(conf), (pos[0] + 4, pos[3] - 4), cv2.FONT_HERSHEY_SIMPLEX, 1, color, 2) return img
def make_window_masked_image(self, img, win_pos, color=utils.BLACK, show_sec=-1): img = utils.draw_box_on_img(img, win_pos, color=color, thickness=-1, alpha=0) utils.imshow(img, desc="make window masked image", pause_sec=show_sec) return img
def make_face_masked_image(img, face_pos_arr, color=utils.RED, margin=0, show_sec=-1): if face_pos_arr is None: return img for face_pos in face_pos_arr: pos = [ face_pos[0] - margin, face_pos[1] - margin, face_pos[2] + margin, face_pos[3] + margin ] pos = utils.check_box_boundary(pos, img.shape[1::-1]) img = utils.draw_box_on_img(img, pos, color=color, thickness=-1, alpha=0) utils.imshow(img, desc="make face masked image", pause_sec=show_sec) return img