Fi5ddlmZddlZddlmZddlZddlZddlm Z m Z m Z ddl m Z ddlmZGddZGd d e Zy) ) annotationsN)Any) BaseSolutionSolutionAnnotatorSolutionResults)LOGGER) check_imshowcPeZdZdZd dZd dZd dZd dZd dZd dZ d dZ d d Z y ) ParkingPtsSelectiona>A class for selecting and managing parking zone points on images using a Tkinter-based UI. This class provides functionality to upload an image, select points to define parking zones, and save the selected points to a JSON file. It uses Tkinter for the graphical user interface. Attributes: tk (module): The Tkinter module for GUI operations. filedialog (module): Tkinter's filedialog module for file selection operations. messagebox (module): Tkinter's messagebox module for displaying message boxes. master (tk.Tk): The main Tkinter window. canvas (tk.Canvas): The canvas widget for displaying the image and drawing bounding boxes. image (PIL.Image.Image): The uploaded image. canvas_image (ImageTk.PhotoImage): The image displayed on the canvas. rg_data (list[list[tuple[int, int]]]): List of bounding boxes, each defined by 4 points. current_box (list[tuple[int, int]]): Temporary storage for the points of the current bounding box. imgw (int): Original width of the uploaded image. imgh (int): Original height of the uploaded image. canvas_max_width (int): Maximum width of the canvas. canvas_max_height (int): Maximum height of the canvas. Methods: initialize_properties: Initialize properties for image, canvas, bounding boxes, and dimensions. upload_image: Upload and display an image on the canvas, resizing it to fit within specified dimensions. on_canvas_click: Handle mouse clicks to add points for bounding boxes on the canvas. draw_box: Draw a bounding box on the canvas using the provided coordinates. remove_last_bounding_box: Remove the last bounding box from the list and redraw the canvas. redraw_canvas: Redraw the canvas with the image and all bounding boxes. save_to_json: Save the selected parking zone points to a JSON file with scaled coordinates. Examples: >>> parking_selector = ParkingPtsSelection() >>> # Use the GUI to upload an image, select parking zones, and save the data c ddl}ddlm}m}td sy|||c|_ |_|_|jj|_ |jjd |jjd d |jj|jd|_|j j#|jj$d|_d|_d|_d|_d|_d|_d|_d|_|jj7|j}|j#|jj8d|j:fd|j<fd|j>ffD]G\}}|jjA|||j#|jjBI|jE|jjGy#t$rCddl}ddddj |j d}tjd |YywxYw)zhInitialize the ParkingPtsSelection class, setting up UI and properties for parking zone point selection.rN) filedialog messageboxzrsudo apt install python3-tk (Debian/Ubuntu) | sudo dnf install python3-tkinter (Fedora) | sudo pacman -S tk (Arch)zgreinstall Python and enable the checkbox `tcl/tk and IDLE` on **Optional Features** during installationzYreinstall Python from https://www.python.org/downloads/macos/ or `brew install python-tk`)LinuxWindowsDarwinz+Unknown OS. Check your Python installation.z8 Tkinter is not configured or supported. Potential fix: T)warnz)Ultralytics Parking Zones Points SelectorFwhite)bg)sidez Upload ImagezRemove Last Bounding BoxSave)textcommand)$tkinterr r ImportErrorplatformgetsystemrwarningr tkTkmastertitle resizableCanvascanvaspackBOTTOMimage canvas_imagecanvas_max_widthcanvas_max_heightrg_data current_boximghimgwFrameTOP upload_imageremove_last_bounding_box save_to_jsonButtonLEFTinitialize_propertiesmainloop) selfrr rr install_cmd button_framercmds j/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/solutions/parking_management.py__init__zParkingPtsSelection.__init__3s  6& 46 J1$/ggjjl  EF eU+ggnnT[[Wn=  dggnn-   $!%   ww}}T[[1 tww{{+T.. / ')F)F G T&& '  YID# GGNN)PILrCrDr askopenfilenamerinfoopenr(sizer/r.minr*intr+r%config PhotoImageresizer) create_imagerNWbindon_canvas_clickr,clearr-)r9rCrDfile aspect_ratio canvas_width canvas_heights r=r2z ParkingPtsSelection.upload_imagensw&..:_9`.a KK, - ZZ% #zz 49yy499, 5AA5EC%%tyy 13tOeOehtOtKu 7Ca6GC&& 2SQ]`lQlMm  mD#..tzz/@/@,P]A^/_`   Adggjj@Q@Q R t';';< d..4466rAc|jj|j|jf|jj |jdz |jdz |jdz|jdzdt |jdk(ri|jj|jj|j|j|jjyy)zCHandle mouse clicks to add points for bounding boxes on the canvas.red)fillN) r-appendxyr% create_ovallenr,copydraw_boxrY)r9events r=rXz#ParkingPtsSelection.on_canvas_clicks %'' 23 ! UWWq[%''A+uwwQR{Y^_ t A % LL   0 0 5 5 7 8 MM$** +    " " $ &rAcztdD]-}|jj||||dzdzdd/y)zADraw a bounding box on the canvas using the provided coordinates.rbrFblue)rarGN)ranger% create_line)r9boxis r=rizParkingPtsSelection.draw_boxsDq TA KK # #CFCQ! ,<6QR # S TrAc|js|jjddy|jj|j y)zARemove the last bounding box from the list and redraw the canvas.WarningzNo bounding boxes to remove.N)r,r showwarningpop redraw_canvasr@s r=r3z,ParkingPtsSelection.remove_last_bounding_boxs=|| OO ' ' 3Q R   rAc|jjd|jjdd|jj|j |j D]}|j|y)z8Redraw the canvas with the image and all bounding boxes.allrrIN)r%deleterUrrVr)r,ri)r9rps r=rvz!ParkingPtsSelection.redraw_canvass\ 5!   Adggjj@Q@Q R<< C MM#  rAcX|j|jjz |j|jj z }}|j Dcgc]3}d|Dcgc]!\}}t ||zt ||zf#c}}i5}}}}ddlm}|}tj||dtddd 5} | j|jd d d |jjd d y cc}}wcc}}}w#1swY3xYw) zMSave the selected parking zone points to a JSON file with scaled coordinates.pointsr)StringIOrb)indentzbounding_boxes.jsonwutf-8encodingNSuccessz+Bounding boxes saved to bounding_boxes.json)r/r% winfo_widthr. winfo_heightr,rQior|jsondumprNwritegetvaluershowinfo) r9scale_wscale_hrprdredatar| write_bufferfs r=r4z ParkingPtsSelection.save_to_jsons99t{{'>'>'@@$))dkkNfNfNhBh^b^j^jkkWZSQTQSW-s1w;/?@QRkkz  $ Q/ 'w ? -1 GGL))+ , -   ,YZRk  - -s$ D*&DD D D D)N)returnNone)rpzlist[tuple[int, int]]rr) __name__ __module__ __qualname____doc__r>r7r2rXrir3rvr4rAr=r r s4 D2hB74%T  [rAr c,eZdZdZdfd ZddZxZS)ParkingManagementa#Manages parking occupancy and availability using YOLO model for real-time monitoring and visualization. This class extends BaseSolution to provide functionality for parking lot management, including detection of occupied spaces, visualization of parking regions, and display of occupancy statistics. Attributes: json_file (str): Path to the JSON file containing parking region details. json (list[dict]): Loaded JSON data containing parking region information. pr_info (dict[str, int]): Dictionary storing parking information (Occupancy and Available spaces). arc (tuple[int, int, int]): BGR color tuple for available region visualization. occ (tuple[int, int, int]): BGR color tuple for occupied region visualization. dc (tuple[int, int, int]): BGR color tuple for centroid visualization of detected objects. Methods: process: Process the input image for parking lot management and visualization. Examples: >>> from ultralytics.solutions import ParkingManagement >>> parking_manager = ParkingManagement(model="yolo26n.pt", json_file="parking_regions.json") >>> print(f"Occupied spaces: {parking_manager.pr_info['Occupancy']}") >>> print(f"Available spaces: {parking_manager.pr_info['Available']}") c nt|d i||jd|_|js t j dt dt|jd5}tj||_dddddd|_ d |_ d |_ d |_ y#1swY)xYw) zVInitialize the parking management system with a YOLO model and visualization settings. json_filezGParkingManagement requires `json_file` with parking region coordinates.u#❌ JSON file path cannot be empty.rrNr) Occupancy Available)rr)rrr)rrr)superr>CFGrrr ValueErrorrNrloadpr_infoarcoccdc)r9kwargsr __class__s r=r>zParkingManagement.__init__s "6"+.~~ NNd eBC C $..7 3 %q ! DI %&'Q7  % %s (B++B4c (|j|t|jd}}t||j}|jD]*}t j |dt jjd}d}t|j|jD]\}} t|d|dzdz t|d|dzdz } } tj|| | fd} | dk\sS|j||j j"t| d d | | d d }n|r |dz }|dz}tj$||gd |r |j&n |j(d -||c|j*d<|j*d<|j-||j*d d d |j/} |j1| t3| |j*d|j*dt|j4S)aProcess the input image for parking lot management and visualization. This function analyzes the input image, extracts tracks, and determines the occupancy status of parking regions defined in the JSON file. It annotates the image with occupied and available parking spots, and updates the parking information. Args: im0 (np.ndarray): The input inference image. Returns: (SolutionResults): Contains processed image `plot_im`, 'filled_slots' (number of occupied parking slots), 'available_slots' (number of available parking slots), and 'total_tracks' (total number of tracked objects). Examples: >>> parking_manager = ParkingManagement(json_file="parking_regions.json") >>> image = cv2.imread("parking_lot.jpg") >>> results = parking_manager.process(image) rr{)dtype)rFrmFrmrFr_)h)rrr T)isClosedcolor thicknessrr)plot_im filled_slotsavailable_slots total_tracks)extract_tracksrgrr line_widthnparrayint32reshapezipboxesclssrQcv2pointPolygonTestdisplay_objects_labelsmodelnames polylinesrrrdisplay_analyticsresultdisplay_outputr track_ids)r9im0roccupied_slots annotatorregionregion_polygonregion_occupiedrpclsxcycinside_distancers r=processzParkingManagement.processs( C *-dii.!%c4??; ii FXXfX&6bhhGOOPZ[N#O DII6 Sc!fs1vo23S#a&3q6/Q9N5OB"%"6"6~BxQV"W"a'44TZZ--c#h7Y[]_ac'+O !#1$ MMn%DHH]a]e]eqr % ,@N< [!4<< #<##C}oWYZ""$ G$k2 LL5T^^,   rA)rrrr)rz np.ndarrayrr)rrrrr>r __classcell__)rs@r=rrs. $; rAr) __future__rrtypingrrnumpyrultralytics.solutions.solutionsrrrultralytics.utilsrultralytics.utils.checksr r rrrAr=rs?#  \\$1`[`[Fe e rA