mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.2 KiB
41 lines
1.2 KiB
#!/usr/bin/python |
|
import cv2.cv as cv |
|
|
|
class PyrSegmentation: |
|
def __init__(self, img0): |
|
self.thresh1 = 255 |
|
self.thresh2 = 30 |
|
self.level =4 |
|
self.storage = cv.CreateMemStorage() |
|
cv.NamedWindow("Source", 0) |
|
cv.ShowImage("Source", img0) |
|
cv.NamedWindow("Segmentation", 0) |
|
cv.CreateTrackbar("Thresh1", "Segmentation", self.thresh1, 255, self.set_thresh1) |
|
cv.CreateTrackbar("Thresh2", "Segmentation", self.thresh2, 255, self.set_thresh2) |
|
self.image0 = cv.CloneImage(img0) |
|
self.image1 = cv.CloneImage(img0) |
|
cv.ShowImage("Segmentation", self.image1) |
|
|
|
def set_thresh1(self, val): |
|
self.thresh1 = val |
|
self.on_segment() |
|
|
|
def set_thresh2(self, val): |
|
self.thresh2 = val |
|
self.on_segment() |
|
|
|
def on_segment(self): |
|
comp = cv.PyrSegmentation(self.image0, self.image1, self.storage, \ |
|
self.level, self.thresh1+1, self.thresh2+1) |
|
cv.ShowImage("Segmentation", self.image1) |
|
|
|
def run(self): |
|
self.on_segment() |
|
cv.WaitKey(0) |
|
|
|
if __name__ == "__main__": |
|
img0 = cv.LoadImage("../c/fruits.jpg", 1) |
|
|
|
# segmentation of the color image |
|
PyrSegmentation(img0).run() |
|
cv.DestroyAllWindows()
|
|
|