0-Ph>ddlZddlZddlmZmZmZ ddl m Z dZ dZ dZd Zee e ee ZeeZd Z ddddZdS)N)polygondiskellipse)warnct|d|dz |d|dz }t|d|dz |d|dz }|d|td|zdz }|d|td|zdz }t|d|d|z|d|z|dg|d|d|d|z|d|zg}d|d|d|zdzf|d|d|zdzfff} || fS)aGenerate a mask for a filled rectangle shape. The height and width of the rectangle are generated randomly. Parameters ---------- point : tuple The row and column of the top left corner of the rectangle. image : tuple The height, width and depth of the image into which the shape is placed. shape : tuple The minimum and maximum size of the shape to fit. random : `numpy.random.Generator` The random state to use for random sampling. Raises ------ ArithmeticError When a shape cannot be fit into the image with the given starting coordinates. This usually means the image dimensions are too small or shape dimensions too large. Returns ------- label : tuple A (category, ((r0, r1), (c0, c1))) tuple specifying the category and bounding box coordinates of the shape. indices : 2-D array A mask of indices that the shape fills. rr rectangle)minintegersmax draw_polygon) pointimageshaperandomavailable_widthavailable_heightrcr labels [/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/skimage/draw/_random_shapes.py_generate_rectangle_maskr sYD%(U1X-uQx8858CO58eAh.a99E!HD a6??3q*:#;#;<< 1 for circles@cannot fit shape to imagecircle) ValueErrorr ArithmeticErrorintr r draw_disk)rrrr min_radius max_radiusleftrighttopbottomavailable_radiusradiusrrs r_generate_circle_maskr+DsS> Qx1}}aA 7888qSJqSJ 8D !HuQx E (C 1Xa F4VZ@@:M!9::: fooc!5E.F.FGGG H HF eAha)6 2 2D  1X  "E!Hv$5 6 1X  "E!Hv$5 6  E ;rct|ddks |ddkrtdt|d|dz |d|d|dz }|d|td|zdz }t t jt jd|z}t|d|d|z |dg|d|d|dzz|d|zg}d|d|z |ddzf|d|d|zdzfff}||fS)aGenerate a mask for a filled equilateral triangle shape. The length of the sides of the triangle is generated randomly. Parameters ---------- point : tuple The row and column of the top left corner of a up-pointing triangle. image : tuple The height, width and depth of the image into which the shape is placed. shape : tuple The minimum and maximum size and color of the shape to fit. random : `numpy.random.Generator` The random state to use for random sampling. Raises ------ ArithmeticError When a shape cannot be fit into the image with the given starting coordinates. This usually means the image dimensions are too small or shape dimensions too large. Returns ------- label : tuple A (category, ((r0, r1), (c0, c1))) tuple specifying the category and bounding box coordinates of the shape. indices : 2-D array A mask of indices that the shape fills. rrz#dimension must be > 1 for trianglesg?rtriangle) rr r r r!npceilsqrtr) rrrravailable_sidesidetriangle_heightr-rs r_generate_triangle_maskr4sOB Qx1}}aA >???qE!H,eAhaAAE!HLN 8fooc!^&<&<== = AD"'"''"2"2T"9::;;O !H !H & !H !H !Htqy !HtO  H  (_ $eAhl 3eAha4RS@S5TU E U?rc x|ddks |ddkrtd|ddz }|ddz }|d}|d|dz }|d}|d|dz } t|||| |} | |krtd||| dz} ||| dz} |tj tj} t |d|d| | |dd| }tjt| | }t j|d}t j |ddz}t j|d}t j |ddz}d ||f||fff}||fS) aGenerate a mask for a filled ellipse shape. The rotation, major and minor semi-axes of the ellipse are generated randomly. Parameters ---------- point : tuple The row and column of the top left corner of the rectangle. image : tuple The height, width and depth of the image into which the shape is placed. shape : tuple The minimum and maximum size and color of the shape to fit. random : `numpy.random.Generator` The random state to use for random sampling. Raises ------ ArithmeticError When a shape cannot be fit into the image with the given starting coordinates. This usually means the image dimensions are too small or shape dimensions too large. Returns ------- label : tuple A (category, ((r0, r1), (c0, c1))) tuple specifying the category and bounding box coordinates of the shape. indices : 2-D array A mask of indices that the shape fills. rrzsize must be > 1 for ellipsesrrNr)rrotationr) rr r uniformr.pi draw_ellipsemathr/r )rrrrr#r$r%r&r'r(r)r_radiusc_radiusr6rmin_xmax_xmin_ymax_yrs r_generate_ellipse_maskrAsB Qx1}}aA 8999qCJqCJ 8D !HuQx E (C 1Xa F4VZ@@*$$9::: ~~j*:Q*>??H~~j*:Q*>??H~~rufbe,,H a aBQBi G3x2233J F71:  E F71:   "E F71:  E F71:   "E %%8 9E E>r)r rr-rc|dkr|f}nt|dkr||z}fd|D}tj|S)aGenerate an array of random colors. Parameters ---------- num_colors : int Number of colors to generate. num_channels : int Number of elements representing color. intensity_range : {tuple of tuples of ints, tuple of ints}, optional The range of values to sample pixel values from. For grayscale images the format is (min, max). For multichannel - ((min, max),) if the ranges are equal across the channels, and ((min_0, max_0), ... (min_N, max_N)) if they differ. random : `numpy.random.Generator` The random state to use for random sampling. Raises ------ ValueError When the `intensity_range` is not in the interval (0, 255). Returns ------- colors : array An array of shape (num_colors, num_channels), where the values for each channel are drawn from the corresponding `intensity_range`. rc^g|])}|d|ddz*S)rr)size)r ).0r num_colorsrs r z+_generate_random_colors..,s7 X X X1fooadAaD1H:o>> X X Xr)lenr. transpose)rF num_channelsintensity_rangercolorss` ` r_generate_random_colorsrM sd:q*, _   " ")L8 X X X X X X X XF <  rFd) channel_axisc  ||dks ||dkrtd|pt|d|d}| d}| |dkrdnd}n5|dkr|fn|} | D]'} | D]"}d|cxkrdksnd}t|#(tj| } |}|d|d|f}tj|dtj }tj|t }g}| ||dz}t|||| }||f}t|D]}|| t}n t|}t| D]}| td|d|z }| td|d|z }||f} ||||| \}}n#t$rg}YwxYw|s||s'||||<d ||<||nt%d | tj|d }ntj|d| }||fS)a Generate an image with random shapes, labeled with bounding boxes. The image is populated with random shapes with random sizes, random locations, and random colors, with or without overlap. Shapes have random (row, col) starting coordinates and random sizes bounded by `min_size` and `max_size`. It can occur that a randomly generated shape will not fit the image at all. In that case, the algorithm will try again with new starting coordinates a certain number of times. However, it also means that some shapes may be skipped altogether. In that case, this function will generate fewer shapes than requested. Parameters ---------- image_shape : tuple The number of rows and columns of the image to generate. max_shapes : int The maximum number of shapes to (attempt to) fit into the shape. min_shapes : int, optional The minimum number of shapes to (attempt to) fit into the shape. min_size : int, optional The minimum dimension of each shape to fit into the image. max_size : int, optional The maximum dimension of each shape to fit into the image. num_channels : int, optional Number of channels in the generated image. If 1, generate monochrome images, else color images with multiple channels. Ignored if ``multichannel`` is set to False. shape : {rectangle, circle, triangle, ellipse, None} str, optional The name of the shape to generate or `None` to pick random ones. intensity_range : {tuple of tuples of uint8, tuple of uint8}, optional The range of values to sample pixel values from. For grayscale images the format is (min, max). For multichannel - ((min, max),) if the ranges are equal across the channels, and ((min_0, max_0), ... (min_N, max_N)) if they differ. As the function supports generation of uint8 arrays only, the maximum range is (0, 255). If None, set to (0, 254) for each channel reserving color of intensity = 255 for background. allow_overlap : bool, optional If `True`, allow shapes to overlap. num_trials : int, optional How often to attempt to fit a shape into the image before skipping it. rng : {`numpy.random.Generator`, int}, optional Pseudo-random number generator. By default, a PCG64 generator is used (see :func:`numpy.random.default_rng`). If `rng` is an int, it is used to seed the generator. channel_axis : int or None, optional If None, the image is assumed to be a grayscale (single channel) image. Otherwise, this parameter indicates which axis of the array corresponds to channels. .. versionadded:: 0.19 ``channel_axis`` was added in 0.19. Returns ------- image : uint8 array An image with the fitted shapes. labels : list A list of labels, one per shape in the image. Each label is a (category, ((r0, r1), (c0, c1))) tuple specifying the category and bounding box coordinates of the shape. Examples -------- >>> import skimage.draw >>> image, labels = skimage.draw.random_shapes((32, 32), max_shapes=3) >>> image # doctest: +SKIP array([ [[255, 255, 255], [255, 255, 255], [255, 255, 255], ..., [255, 255, 255], [255, 255, 255], [255, 255, 255]]], dtype=uint8) >>> labels # doctest: +SKIP [('circle', ((22, 18), (25, 21))), ('triangle', ((5, 6), (13, 13)))] rrz3Minimum dimension must be less than ncols and nrowsNr)rSz1Intensity range must lie within (0, 255) interval)dtypeTzJCould not fit any shapes to image, consider reducing the minimum dimensionr)axisrP)rr r.r default_rngfulluint8zerosboolr rMrangechoice SHAPE_CHOICESSHAPE_GENERATORSr anyappendrsqueezemoveaxis) image_shape max_shapes min_shapesmin_sizemax_sizerJrrK allow_overlap num_trialsrngrQtmpintensity_pair intensitymsg user_shaperfilledlabels num_shapesrL shape_idxshape_generator_columnrowrindicesrs r random_shapesr{0s~+a.  H{1~$=$=NOOO>3{1~{1~>>H &2a&7&7(([$0A$5$5  ?! * *N+ * * Y----#----MC$S//). * )   $ $CJq>;q><@K GKBH 5 5 5E Xk . . .F Fj*q.99J $ZPS T TF x E:&&  !jj77OO.z:Oz""  A\\#aQ()B"C"CDDF,,s1k!nx&?@@AAC&ME !0 UC!P!P"      F7O$7$7$9$9 !' !2g"&w e$$$   :     5q))) E2|44 &=sG'' G65G6) rrNrNNNFrON)r:numpyr.drawrrrr"rr9 _shared.utilsrrr+r4rAdictr`listvaluesr_rMr{rrrs2 UUUUUUUUUU 888v;;;|777tBBBL4& $ "  %,,..// " " " P    [[[[[[[[r