segment.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	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