opencv （python）中没有createGraphSegmentation方法

原因

基于图的分割算法graphsementation没有在opencv主模块版本中实现

解决

使用pip安装 opencv-contrib-python 即可。下面是代码（使用清华镜像源）

pip install opencv-contrib-python -i https://pypi.tuna.tsinghua.edu.cn/simple

安装完成调用cv2.ximgproc.segmentation.createGraphSegmentation()使用

测试代码

（将需要分割的图片重命名为image.jpg）

import numpy as np
import cv2 as cv
import random
#需要图像分割的图片
src = cv.imread('image.jpg')
# 调用方法
segmentator = cv.ximgproc.segmentation.createGraphSegmentation(sigma=0.7, k=500, min_size=5000)

segment = segmentator.processImage(src)
seg_image = np.zeros(src.shape, np.uint8)

for i in range(np.max(segment)):
  # 將第 i 個分割的座標取出
  y, x = np.where(segment == i)

  # 隨機產生顏色
  color = [random.randint(0, 255), random.randint(0, 255),random.randint(0, 255)]

  # 設定第 i 個分割區的顏色
  for xi, yi in zip(x, y):
    seg_image[yi, xi] = color

# 將原始圖片與分割區顏色合併
result = cv.addWeighted(src, 0.3, seg_image, 0.7, 0)

# 顯示結果
cv.imshow("Result", result)
cv.waitKey(0)
cv.destroyAllWindows()