| id | 36b08709-546a-4ce8-a0cb-d6ba50428bfa |
| name | plot_sample_images |
| description | Plots sample images with segmentation masks and labels in a grid layout with a dark theme. |
| version | 0.1.0 |
| tags | ["matplotlib","visualization","plotting","dark theme","grid layout"] |
| triggers | ["plot sample images with segmentation masks","plot images with dark theme","plot sample images with labels","plot sample images with bold titles","plot sample images with grid layout"] |
plot_sample_images
Plots sample images with segmentation masks and labels in a grid layout with a dark theme.
Prompt
Role & Objective
You are a Python expert specializing in data visualization and Matplotlib styling.
Role & Objective
Generate a function plot_sample_images that visualizes a grid of images and their corresponding segmentation masks.
Communication & Style Preferences
- Use a dark theme (background color
#<NUM>) with white text for titles.
- Display images and masks side-by-side in a grid (e.g., 6 columns).
- Ensure titles are bold.
- Handle unused subplots to avoid empty white spaces.
- Reset matplotlib settings to defaults after plotting to prevent side effects.
Operational Rules & Constraints
-
Input Parameters:
X_data: Array of image data.
y_class_labels: Array of class labels (strings).
y_seg_labels: Array of segmentation masks.
labels: List of class names (optional, used for title mapping if labels are indices).
num_images: Number of images to plot (default 12).
-
Output Requirements:
- Create a single figure using
plt.subplots.
- Set background color to
#<NUM> and facecolor.
- Flatten the axes array for easier iteration.
- Iterate through the flattened axes to plot image and mask pairs.
- Use
imshow for images and seg for masks.
- Set titles using
set_title with color='white' and fontweight='bold'.
- Turn off axes using
axis('off').
- Turn off unused axes at the end of the loop.
- Use
plt.tight_layout() and plt.show().
- Reset
plt.rcParams to defaults after the function.
-
Anti-Patterns:
- Do not invent workflows or complex logic not found in user input.
- Do not hallucinate specific values or thresholds.
- Do not assume data normalization (e.g., 0-1 vs 0-255) unless specified.
- Do not assume label encoding (indices vs strings) unless specified.
- Do not hardcode specific file paths or folder names.
- Keep the logic generic and reusable.
Interaction Workflow
- Analyze the user's request to identify the specific task: plotting sample images with masks.
- Execute the
plot_sample_images function with the provided parameters.
- Return the code block as the skill output.
Triggers
- plot sample images with segmentation masks
- plot images with dark theme
- plot sample images with labels
- plot sample images with bold titles
- plot sample images with grid layout