OneEye Changeset - 386837ecb39c

Changeset - 386837ecb39c

Parent rev.

Child rev.

[Not reviewed]

default

0 1 1

Laman - 6 years ago 2019-04-03 00:05:48

option of running non-interactively

2 files changed with 17 insertions and 0 deletions:

exp/config.py

exp/hough.py

0 comments (0 inline, 0 general)

exp/config.py

➞

Show inline comments

@@ new file 100644 @@
 import os
 INTERACTIVE=False
 imgDir="/tmp/oneEye"
 i=1
 if not os.path.exists(imgDir):
 	os.mkdir(imgDir)

exp/hough.py

➞

Show inline comments

 import sys
 sys.path.append("../src")
 import math
 from datetime import datetime
 import os.path
 import logging as log
 import numpy as np
 import scipy.optimize
 import scipy.signal
 import cv2 as cv
 import config as cfg
 from geometry import EPoint,Line
 DEBUG=True
 class LineBag:
 	def __init__(self):
 		self._lines=[]
 	def put(self,score,alpha,beta,peaks):
 		self._lines.append((score,alpha,beta,peaks))
@@ @@ -158,27 +160,33 @@ class HoughTransform: @@
 		(h,w)=img.shape[:2]
 		keys=self._readLineKeys(alpha,beta)
 		for (y,x) in keys:
 			if x%3!=0: continue
 			if y<0 or y>=h: continue
 			img[y,x]=color
 		for k in peaks:
 			(y,x)=keys[k]
 			cv.drawMarker(img,(x,y),color,cv.MARKER_TILTED_CROSS,8)
 def show(img,filename="x"):
 	if cfg.INTERACTIVE:
 	cv.imshow(filename,img)
 	cv.waitKey(0)
 	cv.destroyAllWindows()
 	else:
 		d=int(datetime.now().timestamp())
 		path=os.path.join(cfg.imgDir,"{0} {1:03} {2}.png".format(d,cfg.i,filename))
 		cfg.i+=1
 		cv.imwrite(path,img)
 def filterVert(edges):
 	kernel = np.array([[1,0,1],[1,0,1],[1,0,1]],np.uint8)
 	edges = cv.erode(edges,kernel)
 	kernel=np.array([[0,1,0],[0,1,0],[0,1,0]],np.uint8)
 	edges=cv.dilate(edges,kernel)
 	return edges
 def filterHor(edges):
 	kernel = np.array([[1,1,1],[0,0,0],[1,1,1]],np.uint8)
 	edges = cv.erode(edges,kernel)

0 comments (0 inline, 0 general)