segment.py
4534f6e606c9cluster_pipeline

segment.py
View Options

			import numpy as np

			def segment(I,method,param):
			sI = I.shape
			Iseg = np.zeros(sI)

			if method == 'variance':
			sP = 5
			thresh = param
			for i in range(sI[0]):
			for j in range(sI[1]):
			p = I[ max(i-sP, 1):min(i+sP, sI[0]), max(j-sP, 1):min(j+sP, sI[1]) ]
			p = 1 / (p.shape[0] * p.shape[1]) * np.square(p-np.mean(p))
			var = np.sum(p)
			if var > thresh:
			Iseg[i,j]=1
			return Iseg