图像分割python实现代码

图像分割python实现代码，运用到了最小生成树、聚类等技术

import datetime
import cv2
from base_class.pic2array import PicArray
from base_class.forst import Forest

class Gray2Color:
    v_father_index = 0
    v_int_index = 1
    flag = -1
    row = 0
    col = 0
    imgArray = None
    v_count = 0

    def __init__（self path pictool forst）:
        if isinstance（pictool PicArray）:
            row = pictool.pixRowCount
            col = pictool.pixColCount
            pix_count = row * col
            img_source = cv2.imread（path）
            wheretocolor = pictool.img_source
            if isinstance（forst Forest）:
                forest_list = forst.forest_list
                self.to_color_tree（row col pix_count img_source wheretocolor forest_list）
        else:
            print（‘参数有误！‘）
            return

    def to_color_tree（self row col pix_count img_source wheretocolor forest_list）:
        start = datetime.datetime.now（）
        rgb_dit = {}
        rgb_info = （（0 0 0） （0 0 255） （0 255 0） （0 255 255） （255 0 0） （255 0 255） （255 255 0） （255 255 255））

        index = 0
        vname = 0
        while vname < pix_count:
            if forest_list[vname][0] < 0:
                rgb_dit[vname] = rgb_info[index]
                index = （index + 1）%8
            vname += 1

        index = 0
        while index < pix_count:
            v_tmp_name = index
            v_level = forest_list[v_tmp_name][0]
            while v_level >= 0:
                v_tmp_name = v_level
                v_level = forest_list[v_tmp_name][0]

            wheretocolor[index // col index % col] = rgb_dit[v_tmp_name]
            index += 1

        end = datetime.datetime.now（）
        print（‘上色时间：‘ end - start）
        self.imgArray = wheretocolor

    def show_color_tree（self）:
        cv2.imshow（‘111‘ self.imgArray）
        cv2.waitKey（0）

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2017-06-22 18:36  ImagApart\
     目录           0  2017-06-22 18:37  ImagApart\.idea\
     文件         355  2017-06-22 18:37  ImagApart\.idea\ImagApart.iml
     目录           0  2017-06-22 18:36  ImagApart\.idea\inspectionProfiles\
     文件         228  2017-06-22 18:36  ImagApart\.idea\inspectionProfiles\profiles_settings.xml
     文件         270  2017-06-22 18:36  ImagApart\.idea\modules.xml
     文件        1170  2017-06-22 18:36  ImagApart\.idea\workspace.xml
     目录           0  2017-06-22 18:37  ImagApart\base_class\
     文件        1915  2017-06-17 15:11  ImagApart\base_class\color_tree.py
     文件        4453  2017-06-22 18:35  ImagApart\base_class\forst.py
     文件        4891  2017-06-17 15:02  ImagApart\base_class\pic2array.py
     文件         307  2017-06-16 19:22  ImagApart\base_class\ver_wight_ver.py
     文件           0  2017-06-22 18:36  ImagApart\base_class\__init__.py
     目录           0  2017-06-22 18:37  ImagApart\base_class\__pycache__\
     文件        2031  2017-06-22 18:37  ImagApart\base_class\__pycache__\color_tree.cpython-35.pyc
     文件        2522  2017-06-22 18:37  ImagApart\base_class\__pycache__\forst.cpython-35.pyc
     文件        3549  2017-06-22 18:37  ImagApart\base_class\__pycache__\pic2array.cpython-35.pyc
     文件         137  2017-06-22 18:37  ImagApart\base_class\__pycache__\__init__.cpython-35.pyc
     目录           0  2017-06-22 18:37  ImagApart\demo\
     文件         867  2017-06-22 18:35  ImagApart\demo\demo.py
     文件           0  2017-06-22 18:36  ImagApart\demo\__init__.py
     目录           0  2017-06-22 18:37  ImagApart\resource\
     文件         241  2017-06-04 23:23  ImagApart\resource\asd.png
     文件        7877  2017-05-20 10:30  ImagApart\resource\baidu.jpg
     文件        2321  2017-06-07 11:31  ImagApart\resource\bw.png
     文件    16313948  2017-06-17 14:58  ImagApart\resource\one.txt
     文件       31413  2017-05-28 20:02  ImagApart\resource\test.jpg
     文件       16570  2017-05-30 22:11  ImagApart\resource\test1.jpg
     文件       22692  2017-05-30 22:11  ImagApart\resource\test2.jpg
     文件       22273  2017-05-30 22:11  ImagApart\resource\test3.jpg
     文件    16313948  2017-06-17 15:40  ImagApart\resource\three.txt
............此处省略4个文件信息