2Vh`ddlZddlmZddlmZddlmZed ddZdZy) N)backend)ops) keras_exportz+keras.visualization.draw_segmentation_masksc|xstj}tj|}t |dk7rt d||dk(r,tj |d}tj |d}tj|d}tj|}tj|js-tj|j} td| |1ttjtj|}| t|} nt!|D cgc]} ||  } } tj"||} tj$| d } tj&||}|d d ddf}tj|}|} t)j | d j+d} tj|j-}t/| | D]4\}}t)j0||j}|dddf||d f<6tj|}tj2|d}|rT|d|z z||zz}tj4| d||}tj2|d}tj|}|Scc} w)aDraws segmentation masks on images. The function overlays segmentation masks on the input images. The masks are blended with the images using the specified alpha value. Args: images: A batch of images as a 4D tensor or NumPy array. Shape should be (batch_size, height, width, channels). segmentation_masks: A batch of segmentation masks as a 3D or 4D tensor or NumPy array. Shape should be (batch_size, height, width) or (batch_size, height, width, 1). The values represent class indices starting from 1 up to `num_classes`. Class 0 is reserved for the background and will be ignored if `ignore_index` is not 0. num_classes: The number of segmentation classes. If `None`, it is inferred from the maximum value in `segmentation_masks`. color_mapping: A dictionary mapping class indices to RGB colors. If `None`, a default color palette is generated. The keys should be integers starting from 1 up to `num_classes`. alpha: The opacity of the segmentation masks. Must be in the range `[0, 1]`. blend: Whether to blend the masks with the input image using the `alpha` value. If `False`, the masks are drawn directly on the images without blending. Defaults to `True`. ignore_index: The class index to ignore. Mask pixels with this value will not be drawn. Defaults to -1. data_format: Image data format, either `"channels_last"` or `"channels_first"`. Defaults to the `image_data_format` value found in your Keras config file at `~/.keras/keras.json`. If you never set it, then it will be `"channels_last"`. Returns: A NumPy array of the images with the segmentation masks overlaid. Raises: ValueError: If the input `images` is not a 4D tensor or NumPy array. TypeError: If the input `segmentation_masks` is not an integer type. z;`images` must be batched 4D tensor. Received: images.shape=channels_first)rfloat32)dtypezR`segmentation_masks` must be in integer dtype. Received: segmentation_masks.dtype=N)axis.r)r rr r )axesboolr ).Nuint8)rimage_data_formatrshapelen ValueError transposeconvert_to_tensor is_int_dtyper standardize_dtype TypeErrorintconvert_to_numpymax_generate_color_paletterange not_equalsqueezeone_hotnpastypecopyziparraycastwhere)imagessegmentation_masks num_classes color_mappingalphablend ignore_index data_format images_shaper colorsi valid_masksmasksimages_to_drawmaskcoloroutputss _/home/dcms/DCMS/lib/python3.12/site-packages/keras/src/visualization/draw_segmentation_masks.pydraw_segmentation_masksr=s`<!:!:!?q-"??-- 2LAK++k3K%7E+CAI6--.@A E LL\ 2 9 9& AE))&1668N5&)3 en&:&:;$)$'NtSy!3**>:Nhh~Y7G AI&58))K 2GVD((7'2&&w/ N-@s Kctjgd}t|Dcgc]}||zdzjc}Scc}w)N)iii)r$r(r tolist)r-paletter5s r<rrks<hh89G49+4F Gqa'kS ( ( * GG GsA)NNg?TrN) numpyr$ keras.srcrrkeras.src.api_exportrr=rr<rGsH-;< _=_DHrF