OneEye Changeset - 28c6f89a3a7e

Changeset - 28c6f89a3a7e

Parent rev.

Child rev.

[Not reviewed]

default

0 1 0

Laman - 6 years ago 2019-02-09 12:59:46

stone detection enhanced by edge subtraction

1 file changed with 12 insertions and 8 deletions:

exp/board_detect.py

0 comments (0 inline, 0 general)

exp/board_detect.py

➞

Show inline comments

@@ @@ -84,49 +84,53 @@ def groupLines(points,minCount,tolerance @@
 class BoardDetector:
 	def __init__(self,annotationsPath):
 		self._annotations=DataFile(annotationsPath)
 		self._rectW=0
 		self._rectH=0
 		self._rect=None
 	def __call__(self,img,filename):
 		# approximately detect the board
 		(h,w)=img.shape[:2]
 		log.debug("image dimensions: %s x %s",w,h)
 		show(img,filename)
 		(x1,y1,x2,y2)=self._detectRough(img,filename)
 		rect=img[y1:y2,x1:x2]
 		self._rectW=x2-x1
 		self._rectH=y2-y1
 		self._rect=rect
 		# quantize colors
 		colors=self._sampleColors(rect)
 		quantized=quantize(rect,colors)
 		show(quantized,filename)
 		gray=cv.cvtColor(rect,cv.COLOR_BGR2GRAY)
 		edges=cv.Canny(gray,70,130)
 		show(edges,"edges")
 		quantized=quantized & (255-cv.cvtColor(edges,cv.COLOR_GRAY2BGR))
 		show(quantized,"quantized, edges separated")
 		# detect black and white stones
 		stones=self._detectStones(quantized,colors)
 		# detect lines from edges and stones
 		edgeImg=prepareEdgeImg(rect)
 		hough=HoughTransform(edgeImg)
 		hough.extract()
 		stonesImg=np.zeros((self._rectH,self._rectW),np.uint8)
 		for (point,c) in stones:
 			cv.circle(stonesImg,(int(point.x),int(point.y)),2,255,-1)
 		# cv.drawContours(stonesImg,[c for (point,c) in stones],-1,255,-1)
 		show(stonesImg,"detected stones")
 		# hough.update(stonesImg,5)
 		hough=HoughTransform(stonesImg)
 		hough.extract()
 		# # detect lines passing through the stones
 		# lines=self._constructLines(stones)
+		#
 		# # detect vanishing points of the lines
 		# imgCenter=EPoint(w//2-x1, h//2-y1)
 		# (a,b,c,d)=(p-EPoint(x1,y1) for p in self._annotations[filename][0])
 		# (p,q,r,s)=(Line(a,b),Line(b,c),Line(c,d),Line(d,a))
@@ @@ -144,58 +148,58 @@ class BoardDetector: @@
 	def _detectRough(self,img,filename):
 		corners=self._annotations[filename][0]
 		(x1,y1,x2,y2)=computeBoundingBox(corners)
 		log.debug("bounding box: (%s,%s) - (%s,%s)",x1,y1,x2,y2)
 		return (x1,y1,x2,y2)
 	def _sampleColors(self,rect):
 		(h,w)=rect.shape[:2]
 		minirect=rect[h//4:3*h//4, w//4:3*w//4]
 		return kmeans(minirect)
 	def _detectStones(self,quantized,colors):
 		(h,w)=quantized.shape[:2]
 		mask=self._maskStones(quantized,colors)
 		stoneDims=(w/19,h/19)
 		log.debug("stone dims: %s - %s",tuple(x/2 for x in stoneDims),stoneDims)
 		(contours,hierarchy)=cv.findContours(mask,cv.RETR_LIST,cv.CHAIN_APPROX_SIMPLE)
 		stoneLocs=filterStones(contours,mask,stoneDims)
 		return stoneLocs
 	def _maskStones(self,quantized,colors):
 		unit=np.array([1,1,1],dtype=np.uint8)
 		kernel=np.ones((3,3),np.uint8)
 		maskB=cv.inRange(quantized,colors[0]-unit,colors[0]+unit)
 		maskB=cv.morphologyEx(maskB,cv.MORPH_OPEN,kernel,iterations=1)
 		maskB=cv.erode(maskB,kernel,iterations=3)
 		# distTransform = cv.distanceTransform(maskB,cv.DIST_L2,5)
 		# maskB=cv.inRange(distTransform,6,10)
 		distTransform=cv.distanceTransform(maskB,cv.DIST_L2,5)
 		maskB=cv.inRange(distTransform,6,20)
 		show(maskB,"black areas")
 		maskW=cv.inRange(quantized,colors[1]-unit,colors[1]+unit)
 		maskW=cv.morphologyEx(maskW,cv.MORPH_OPEN,kernel,iterations=1)
 		maskW=cv.erode(maskW,kernel,iterations=2)
 		distTransform=cv.distanceTransform(maskW,cv.DIST_L2,5)
 		maskW=cv.inRange(distTransform,6,20)
 		show(maskW,"white areas")
 		stones=cv.bitwise_or(maskB,maskW)
 		show(stones,"black and white areas")
 		return stones
 	def _constructLines(self,stoneLocs):
 		lineDict=dict()
 		# minCount=min(max(math.sqrt(len(stoneLocs))-4,3),7)
 		minCount=6
 		log.debug("min count: %s",minCount)
 		points=[point for (point,contour) in stoneLocs]
 		for line in groupLines(points,minCount,2):
 			key=line.getSortedPoints()
 			if key in lineDict: # we already have a line with the same incident points
 				continue
 			lineDict[line.getSortedPoints()]=line
 			obsolete=set()
 			for ab in lineDict.values():
 				if ab is line: continue
 				if line.points<ab.points: # == impossible
 					del lineDict[key]
 					break
 				if ab.points<line.points:
 					obsolete.add(ab.getSortedPoints())

0 comments (0 inline, 0 general)