Changeset - 8b30e6dba468
[Not reviewed]
default
0 1 0
Laman - 6 years ago 2019-05-05 17:55:18

fix: corrupted training data
1 file changed with 2 insertions and 2 deletions:
0 comments (0 inline, 0 general)
exp/kerokero/prepare_data.py
Show inline comments
 
@@ -78,14 +78,14 @@ def harvestDir(path):
 
	boards=annotations["."]
 
	for f in files:
 
		img=cv.imread(f.path)
 
		img=cv.cvtColor(img,cv.COLOR_BGR2GRAY)
 
		for b in boards:
 
			sample=Sample(img,b.grid)
 
			(img,label)=sample.transform()
 
			yield (img,label)
 
			(transformedImg,label)=sample.transform()
 
			yield (transformedImg,label)
 

	
 

	
 
def loadDataset(root):
 
	testRatio=0.1
 
	trainRatio=1-testRatio
 
	images=[]
0 comments (0 inline, 0 general)