a #Df}@s4ddlZddlZddlmZdddZd ddZdS) N)utilscCs(t|}t||||||dt|S)aDraw rectangle on numpy array with Pillow. Parameters ---------- src: numpy.ndarray Input image. aabb1: array-like, (2,) Minimum vertex (y_min, x_min) of the axis aligned bounding box (AABB). aabb2: array-like, (2,) Maximum vertex (y_max, x_max) of the AABB. fill: int or array-like, (3,), optional RGB color to fill the mark. None for no fill. (default: None) outline: int or array-like, (3,), optional RGB color to draw the outline. width: int, optional Rectangle line width. (default: 0) Returns ------- dst: numpy.ndarray Output image. )imgaabb1aabb2filloutlinewidth)rZnumpy_to_pillow rectangle_Zpillow_to_numpy)srcrrrrr dstr b/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/imgviz/draw/rectangle.py rectangles rc Csft|tjjrt|}t|tjjr,t|}tj|}|\}}|\} } |j||| | f|||ddS)N)Zxyrrr ) isinstance collectionsabcIterabletuplePILZ ImageDrawr) rrrrrr Zdrawy1x1y2Zx2r r rr ,s r )NNr)NNr)rZ PIL.Imagerrrr r r r rs  $