del draw ImageShow.show(skewless_picture, 'with points') skewless_picture.save('./images/intermediate/' + pic_name + '_foundnumbers.jpg') # save training images if TRAIN: for point in positions: ptx = int(point[0]) pty = int(point[1]) picture = skewless_picture.crop([ptx, pty, ptx + puzzlegrid.cell_width, pty + puzzlegrid.cell_height]); picture.save("./images/train/{0}_{1}_{2}.jpg".format(pic_name, ptx, pty)) digits = Representations.DigitsRep(top, side) print "top" print digits.top print "side" print digits.side return digits # digits representation # testing main if __name__=='__main__': test_pic = './images/blue_9.jpg' image = Image.open(test_pic) puzzlegrid = DetectGrid.detect(image) nums = find_numbers(puzzlegrid, image) print nums
'_foundnumbers.jpg') # save training images if TRAIN: for point in positions: ptx = int(point[0]) pty = int(point[1]) picture = skewless_picture.crop([ ptx, pty, ptx + puzzlegrid.cell_width, pty + puzzlegrid.cell_height ]) picture.save("./images/train/{0}_{1}_{2}.jpg".format( pic_name, ptx, pty)) digits = Representations.DigitsRep(top, side) print "top" print digits.top print "side" print digits.side return digits # digits representation # testing main if __name__ == '__main__': test_pic = './images/blue_9.jpg' image = Image.open(test_pic) puzzlegrid = DetectGrid.detect(image) nums = find_numbers(puzzlegrid, image) print nums
import Image import GetPicture import AntiSkew import DetectGrid import FindNumbers import OCRNumbers import PicrossSolver import DisplayPuzzle pic_name = 'snapshot' test_pic = './images/' + pic_name + '.jpg' # main if __name__=='__main__': <<<<<<< HEAD original_picture = GetPicture.picture(pic_name, test_pic) #original_picture = Image.open(test_pic) skewless_picture = AntiSkew.fix_skew(original_picture) ======= #original_picture = GetPicture.picture(pic_name, test_pic) original_picture = Image.open(test_pic) skewless_picture = AntiSkew.fix_skew(pic_name, original_picture) >>>>>>> 0d327375a183089fefc8a7d1792d145aa8646a50 puzzlegrid = DetectGrid.detect_lines(skewless_picture,"test") number_positions = FindNumbers.find_numbers(puzzlegrid, skewless_picture, pic_name) #puzzle = OCRNumbers.ocr(number_positions, skewless_picture, puzzlegrid) #solution = PicrossSolver.solve(puzzle) #DisplayPuzzle.visualize(solution)