o bi@srddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ed     dd d Z dS)N)backend)ops) keras_export)draw_segmentation_masks)plot_image_galleryz2keras.visualization.plot_segmentation_mask_galleryrT皙?c Ksz| pt} t|} t| dkrtd| | dkr"t|d}t| dkr,| dnd} | } d}|dur:|d7}|durB|d7}t|}tj t |||||d}|durk| dkr`t|d}t |d }|||}|dur| dkryt|d}t |d }|||}g}t | D]}| |||dur| |||dur| ||qtj|dd }t|f|| |d | S) aPlots a gallery of images with corresponding segmentation masks. Args: images: A 4D tensor or NumPy array of images. Shape should be `(batch_size, height, width, channels)`. num_classes: The number of segmentation classes. Class indices should start from `1`. Class `0` will be treated as background and ignored if `ignore_index` is not 0. value_range: A tuple specifying the value range of the images (e.g., `(0, 255)` or `(0, 1)`). Defaults to `(0, 255)`. y_true: A 3D/4D tensor or NumPy array representing the ground truth segmentation masks. Shape should be `(batch_size, height, width)` or `(batch_size, height, width, 1)`. Defaults to `None`. y_pred: A 3D/4D tensor or NumPy array representing the predicted segmentation masks. Shape should be the same as `y_true`. Defaults to `None`. color_mapping: A dictionary mapping class indices to RGB colors. If `None`, a default color palette is used. Class indices start from `1`. Defaults to `None`. 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`. alpha: The opacity of the segmentation masks (a float between 0 and 1). Defaults to `0.8`. ignore_index: The class index to ignore when drawing masks. Defaults to `-1`. data_format: The image data format `"channels_last"` or `"channels_first"`. Defaults to the Keras backend data format. kwargs: Additional keyword arguments to be passed to `keras.visualization.plot_image_gallery`. Returns: The output of `keras.visualization.plot_image_gallery`. Raises: ValueError: If `images` is not a 4D tensor/array. z;`images` must be batched 4D tensor. Received: images.shape=channels_first)rrrN) num_classes color_mappingalpha ignore_indexblendint32)axis) value_rangerowscols)rimage_data_formatrshapelen ValueError transposeconvert_to_numpy functoolspartialrcastrangeappendnpstackr)imagesrry_truey_predrrrr data_formatkwargs image_shape batch_sizerr images_np draw_masks_fntrue_masks_drawnpredicted_masks_drawnimages_with_masksigallery_imagesr5j/home/ubuntu/.local/lib/python3.10/site-packages/keras/src/visualization/plot_segmentation_mask_gallery.pyplot_segmentation_mask_gallerysh 3            r7)rNNNTr r N) r numpyr% keras.srcrrkeras.src.api_exportr/keras.src.visualization.draw_segmentation_masksr*keras.src.visualization.plot_image_galleryrr7r5r5r5r6s"