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)
