def guessedWrong(originalImage):
print 'WHOOPS!'
cont = True
while (cont):
correctDigit = raw_input("--> Please input the correct digit ('X' to exit): ")
if correctDigit == 'X' or correctDigit == 'x':
cont = False
else:
if utilities.isInteger(correctDigit):
for i in range (0,10):
if correctDigit == "{0}".format(i):
saveImageToCorrectDigitFolder(originalImage,correctDigit)
print "Please go back to the main menu and restart the Digit Recognition program to reload the new image into the database!"
cont = False
if cont:
print "Command must be an integer from [0,10], or 'X' to exit!"
else:
print "Command must be an integer from [0,10], or 'X' to exit!"
return
def filterFunction(originalImage):
print 'Kernel Types:'
for kernelIndex in range(0,len(kernels.kernels)):
print '({0}) {1}'.format(kernelIndex, kernels.kernels[kernelIndex]['name'])
print '({0}) ZS Thinning'.format(len(kernels.kernels))
cont = True
while (cont):
command = raw_input('--> Please choose a filter: ')
if (utilities.isInteger(command)):
command = int(command)
if (command >= 0 and command <= (len(kernels.kernels))):
cont = False
else:
print "Must be an integer in the range [0, {0}]!".format(len(kernels.kernels))
else:
print "Must be an integer in the range [0, {0}]!".format(len(kernels.kernels))
if command == len(kernels.kernels):
print 'Thinning...'
thinnedImage = thinningFilter(originalImage)
thinnedImage.show()
print 'Saving the thinned image to the output folder as thinnedImage.jpg...'
thinnedImage.save(os.getcwd() + '/output/thinnedImage.jpg')
else:
multiplier = 1.0/float(kernels.kernels[command]['factor'])
print 'Applying the filter - {0}...'.format(kernels.kernels[command]['name'])
filteredImage = apply3x3Filter(originalImage,kernels.kernels[command]['kernel'],multiplier)
# filteredImage = Image.fromarray(ndimage.convolve(numpy.asarray(originalImage),kernels.kernels[command]['kernel']))
print 'Displaying the image with the applied filter - {0}...'.format(kernels.kernels[command]['name'])
filteredImage.show()
print 'Saving the image with the applied filter to the output folder as filteredImage.jpg...'
filteredImage.save(os.getcwd() + '/output/filteredImage.jpg')
return
