Cleaned up image files

src/machine_learning/custom_scripts/image_augment.py

@@ -13,6 +13,7 @@
 import csv
 import ast
 
+Image.Image.tostring = Image.Image.tobytes
 
 def parse(ifile, ofile):
 	'''
@@ -104,7 +105,7 @@ def main(argv):
 				output_img_directory = arg
 	else:
 		print 'image_augment.py -i <training_csv_file> -o <output_csv_file> -I <input_img_directory>  -O <out_img_directory>'
-		sys.exit2()
+		sys.exit(2)
 
 	if not parse(inputfile, outputfile):
 		print "Please enter valid CSV file(s)"
@@ -200,4 +201,4 @@ def main(argv):
 	else:
 		print "Not found"
 if __name__ == "__main__":
-	main(sys.argv[1:])
+	main(sys.argv[1:])