OneEye Changeset - 9c78e6f2e2ea

Changeset - 9c78e6f2e2ea

Parent rev.

Child rev.

[Not reviewed]

default

0 2 0

Laman - 6 years ago 2019-05-14 10:28:31

fixed data generation, color sampler showing filename

2 files changed with 5 insertions and 4 deletions:

exp/color_sampler.py

exp/kerokero/prepare_data.py

0 comments (0 inline, 0 general)

exp/color_sampler.py

➞

Show inline comments

@@ @@ -39,25 +39,27 @@ class Sampler: @@
 			return
 		(h,s,v)=map(lambda x: round(x,3), rgb_to_hsv(r,g,b))
 		self.hsv=(h,s,v)
 		self.pos=(e.x,e.y)
 		self.posLabel.configure(text="({0},{1})".format(e.x,e.y))
 		self.colorLabel.configure(text="HSV: ({0}, {1}, {2})".format(round(h*360),s,v))
 	def printSample(self):
 		print("\t".join(map(str, (self.filenames[self.k],*self.pos,self.letter,*self.hsv))))
 	def showImage(self):
 		self.img=Image.open(os.path.join(self.dirname,self.filenames[self.k]))
 		filename=self.filenames[self.k]
 		self.img=Image.open(os.path.join(self.dirname,filename))
 		self.root.title(filename)
 		(w,h)=self.img.size
 		self.photo=ImageTk.PhotoImage(self.img)
 		self.canvas.delete("all")
 		self.canvas.create_image(0,0,image=self.photo,anchor="nw")
 		self.canvas.configure(width=w,height=h)
 		self.m=0
 		self.isGrid=False
 		f=self.filename
 		if f in self.annotations:
 			self._boards=self.annotations[f]+[Board()]
 		else:
@@ @@ -101,25 +103,25 @@ class Sampler: @@
 		self._dirty=True
 		self._markCorners()
 	def save(self):
 		if self._dirty:
 			self.annotations.save()
 			self._dirty=False
 	def _flush(self):
 		self.annotations[self.filename]=[b for b in self._boards if not b.isEmpty()]
 	def _createGUI(self):
 		root=tk.Tk()
+		self.root=root=tk.Tk()
 		frame=tk.Frame(root)
 		frame.grid(column=0,row=0,sticky=(N,S,E,W))
 		self.canvas=tk.Canvas(frame)
 		self.canvas.grid(row=1,column=0)
 		bar=self._createBar(frame)
 		bar.grid(row=2,column=0,sticky=(E,W))
 		self.showImage()
 		self.canvas.bind('<1>',lambda e: self.printSample())

exp/kerokero/prepare_data.py

➞

Show inline comments

 import os
 import sys
 import re
 import random
 import itertools
 import numpy as np
 import cv2 as cv
 sys.path.append("..")
 sys.path.append("../../src")
 from annotations import DataFile,computeBoundingBox,Corners,EPoint
 from geometry import Line
 from kerokero.transformation_matrices import getIdentity,getRotation,getTranslation,getScale,getMirroring,getProjection
 random.seed(361)
@@ @@ -31,25 +30,25 @@ class Sample: @@
 		center=self._getCenter()
 		m=getIdentity()
 		t1=getTranslation(-center.x,-center.y)
 		proj=getProjection()
 		rot=getRotation()
 		mir=getMirroring()
 		for mi in [t1,mir,proj,rot]:
 			m=np.matmul(mi,m)
 		m=np.matmul(self._computeCrop(m),m)
 		img=cv.warpPerspective(self.img,m,(self.SIDE,self.SIDE))
 		img=np.float32(img)/128-1
 		grid=Corners(c.transform(m) for c in self.grid)
 		grid=list(map(lambda p: 2*p/self.SIDE-EPoint(1,1), grid))
+		grid=list(map(lambda p: list(2*p/self.SIDE-EPoint(1,1)), grid))
 		return (img,grid)
 	def _getCenter(self):
 		(a,b,c,d)=self.grid
 		p=Line.fromPoints(a,c)
 		q=Line.fromPoints(b,d)
 		return p.intersect(q)
 	def _computeCrop(self,m):
 		grid=Corners(c.transform(m) for c in self.grid)
 		(x1,y1,x2,y2)=computeBoundingBox(grid)
 		(wg,hg)=(x2-x1,y2-y1)

0 comments (0 inline, 0 general)