from skimage.segmentation import slic from skimage.segmentation import mark_boundaries from skimage.util import img_as_float import matplotlib.pyplot as plt import numpy as np import cv2 # args args = {"image": 'I:\\18Breakageratecalculation\\SVM run\\images\\030.jpg'} # load the image and apply SLIC and extract (approximately) # the supplied number of segments image = cv2.imread(args["image"]) segments = slic(img_as_float(image), n_segments=100, sigma=3) # show the output of SLIC fig = plt.figure('Superpixels') ax = fig.add_subplot(1, 1, 1) ax.imshow(mark_boundaries(img_as_float(cv2.cvtColor(image, cv2.COLOR_BGR2RGB)), segments)) plt.axis("off") plt.show() print("segments:\n", segments) print("np.unique(segments):", np.unique(segments)) # loop over the unique segment values for (i, segVal) in enumerate(np.unique(segments)): # construct a mask for the segment print("[x] inspecting segment {}, for {}".format(i, segVal)) mask = np.zeros(image.shape[:2], dtype="uint8") mask[segments == segVal] = 255 # show the masked region cv2.imshow("Mask", mask) cv2.imshow("Applied", np.multiply(image, cv2.cvtColor(mask, cv2.COLOR_GRAY2BGR) > 0)) cv2.waitKey(0) cv2.imwrite("segment_%d.png" % i, np.multiply(image, cv2.cvtColor(mask, cv2.COLOR_GRAY2BGR) > 0))导出子块图像到指定文件夹

可以使用以下代码将每个超像素块保存为单独的图像文件:

import os

# create a directory to store the segmented images
if not os.path.exists("segmented_images"):
    os.makedirs("segmented_images")

# loop over the unique segment values
for (i, segVal) in enumerate(np.unique(segments)):
    # construct a mask for the segment
    mask = np.zeros(image.shape[:2], dtype="uint8")
    mask[segments == segVal] = 255     

    # apply the mask to the image
    masked_image = np.multiply(image, cv2.cvtColor(mask, cv2.COLOR_GRAY2BGR) > 0)

    # save the masked image as a file
    filename = os.path.join("segmented_images", "segment_%d.png" % i)
    cv2.imwrite(filename, masked_image)

这将在当前目录下创建一个名为 "segmented_images" 的文件夹，并将每个超像素块保存为单独的图像文件。