def __init__(self, directory, CELLS_PER_ROW, CELLS_PER_COLUMN): self.directory = directory self.images = [] self.cpr = CELLS_PER_ROW self.cpc = CELLS_PER_COLUMN self.histograms = [[numpy.zeros(256) for i in xrange(self.cpc)] for i in xrange(self.cpr)] jpgs = glob.glob("%s/*.jpg" % self.directory) print "Found %s images" % len(jpgs) default = pImage(jpgs[0]) default.makeDefault() default.cellify(self.cpr, self.cpc) self.generateLuminance(default) default.classifyCells(True) default.setBoardArea() default.free() self.images = [0 for x in xrange(len(jpgs))] for index in xrange(len(jpgs)): pImg = pImage(jpgs[index]) pImg.cellify(self.cpr, self.cpc) self.images[index] = pImg
def getStartingKeyImage(self): keyImage = pImage(self.images[0].filename) keyImage.cellify(self.cpr, self.cpc) keyImage.classifyCells(True) return keyImage