| import os | |
| import numpy as np | |
| import argparse | |
| from closed_eyes_detection import run_on_webcam, ClosedEyesDetector | |
| from blurriness_estimation import BlurrinessEstimator | |
| from config import BlurrinessConfig, InferenceConfig | |
| import cv2 | |
| def parse_arguments() -> argparse.ArgumentParser: | |
| """ | |
| Parse the cli arguments given by the user. | |
| Returns: | |
| parser: parser object containing arguments and their values | |
| """ | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("-w", "--webcam", required=False, action="store_true", default=False, | |
| help="Runs eye blink detection on the user's webcam. Overrides other options.") | |
| parser.add_argument("-p", "--path", required=False, type=str, | |
| help="path of image to load") | |
| parser.add_argument("-e", "--eyes", required=False, action="store_true", default=False, | |
| help="Detects eye blink on the input image") | |
| parser.add_argument("-b", "--blur", required=False, action="store_true", default=False, | |
| help="Estimates image blur") | |
| return parser | |
| def detect_closed_eyes(img: np.ndarray) -> str: | |
| """ | |
| Runs the Closed Eyes Detector | |
| Args: | |
| img: input image | |
| Returns: | |
| str: Whether the eyes are opened or closed. | |
| """ | |
| detector = ClosedEyesDetector(InferenceConfig()) | |
| is_closed = detector(img) | |
| s = "Eyes are closed." if is_closed else "Eyes are open." | |
| return s | |
| def detect_blur(img: np.ndarray) -> str: | |
| """ | |
| Runs the Blurriness Detector | |
| Args: | |
| img: input image | |
| Returns: | |
| str: Whether the input image is blurry | |
| """ | |
| cfg = BlurrinessConfig() | |
| be = BlurrinessEstimator(cfg) | |
| is_blurry, score = be(img) | |
| s = "Image is blurry" if is_blurry else "Image is not blurry" | |
| s += f" - Score: {score:.0f} - Sharpness threshold: {cfg.threshold}" | |
| return s | |
| def main(): | |
| parser = parse_arguments() | |
| args = vars(parser.parse_args()) | |
| if args['webcam']: | |
| run_on_webcam() | |
| return | |
| if args['path'] is None: | |
| print('Image path required (-p).') | |
| parser.print_help() | |
| else: | |
| path = args['path'] | |
| if not os.path.exists(path): | |
| print(f"File not found at {path}") | |
| return | |
| img = cv2.imread(path) | |
| img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) | |
| if args['eyes']: | |
| result = detect_closed_eyes(img) | |
| elif args['blur']: | |
| result = detect_blur(img) | |
| else: | |
| print('No task specified. Use either -b or -e.') | |
| parser.print_help() | |
| return | |
| print(result) | |
| if __name__ == "__main__": | |
| main() | |