def getListOfDotCoords(self, imagePath, dotImagesPath):
t = Trainer()
e = Extractor()
v = Vectors()
imageTemplateList = t.createListOfImageTemplates("dot", dotImagesPath)
pts = sorted(e.getMatchingPoints(imagePath, imageTemplateList, 0.9)[0])
for pt in pts:
for pt2 in pts[pts.index(pt):]:
if pt == pt2:
continue
else:
if v.distance(pt, pt2) < 10:
pts.remove(pt2)
return pts
import sys
sys.path.insert(0, '../src/modules/Feature_Extractor')
from Train import Trainer
from Loc_Extractor import Extractor
e = Extractor()
t = Trainer()
t.createIntervalFrameImages("../data/object_videos/cube1.mov", "out_dir", "cube","1")
#t.createIntervalFrameImages("../data/object_videos/sphere1.mov", "out_dir", "sphere","1")
#t.createIntervalFrameImages("../data/object_videos/sphere2.mov", "out_dir", "sphere","2")
cubeTemplateList = t.createListOfImageTemplates("cube", "out_dir")
#templateListList = [cubeTemplateList,sphereTemplateList]
print e.getObjectLoc("../data/image.png", cubeTemplateList, 0.7)
#e.getObjectLoc( "../data/cube.png", cubeTemplateList, 0.8)
#print e.findAllObjects("../data/both_alt_env.png",templateListList, 0.8)