o W h @s4ddlmZmZmZddlmZGdddeZdS)) BaseSolutionSolutionAnnotatorSolutionResults)colorscs(eZdZdZfddZddZZS) VisionEyea A class to manage object detection and vision mapping in images or video streams. This class extends the BaseSolution class and provides functionality for detecting objects, mapping vision points, and annotating results with bounding boxes and labels. Attributes: vision_point (Tuple[int, int]): Coordinates (x, y) where vision will view objects and draw tracks. Methods: process: Process the input image to detect objects, annotate them, and apply vision mapping. Examples: >>> vision_eye = VisionEye() >>> frame = cv2.imread("frame.jpg") >>> results = vision_eye.process(frame) >>> print(f"Total detected instances: {results.total_tracks}") c s$tjdi||dd|_dS)z Initialize the VisionEye class for detecting objects and applying vision mapping. Args: **kwargs (Any): Keyword arguments passed to the parent class and for configuring vision_point. vision_point)rN)super__init__getr)selfkwargs __class__r T/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/solutions/vision_eye.pyr szVisionEye.__init__cCs||t||j}t|j|j|j|jD]\}}}}|j|| |||t t |dd| ||j q|}||t|t|jdS)a Perform object detection, vision mapping, and annotation on the input image. Args: im0 (numpy.ndarray): The input image for detection and annotation. Returns: (SolutionResults): Object containing the annotated image and tracking statistics. - plot_im: Annotated output image with bounding boxes and vision mapping - total_tracks: Number of tracked objects in the frame Examples: >>> vision_eye = VisionEye() >>> frame = cv2.imread("image.jpg") >>> results = vision_eye.process(frame) >>> print(f"Detected {results.total_tracks} objects") T)labelcolor)plot_im total_tracks)extract_tracksr line_widthzipclss track_idsboxesconfs box_labeladjust_box_labelrint visioneyerresultdisplay_outputrlen)r im0 annotatorclst_idboxconfrr r rprocess's  "$ zVisionEye.process)__name__ __module__ __qualname____doc__r r* __classcell__r r rrrs  rN)ultralytics.solutions.solutionsrrrultralytics.utils.plottingrrr r r rs