Classify an image from a local file path.
(file_path)
| 107 | |
| 108 | |
| 109 | def classify_local_image(file_path): |
| 110 | """ Classify an image from a local file path. """ |
| 111 | if not file_path.endswith('.jpg') and not file_path.endswith('.png'): |
| 112 | raise Exception("Expecting a .jpg or .png file!") |
| 113 | downloads_folder = settings.DOWNLOADS_FOLDER |
| 114 | hex_name = 'temp_image_%s' % uuid.uuid4().hex |
| 115 | hex_name_png = hex_name + '.png' |
| 116 | hex_name_jpg = hex_name + '.jpg' |
| 117 | shutil.copy2(file_path, os.path.join(downloads_folder, hex_name_png)) |
| 118 | convert_image_file_to_jpg( |
| 119 | "%s/%s" % (downloads_folder, hex_name_png)) |
| 120 | os.rename(downloads_folder + "/" + hex_name_png, |
| 121 | downloads_folder + "/temp_image_png.png") |
| 122 | best_guess = classify_image.external_run( |
| 123 | "%s/%s" % (downloads_folder, hex_name_jpg)) |
| 124 | os.rename(downloads_folder + "/" + hex_name_jpg, |
| 125 | downloads_folder + "/temp_image_jpg.jpg") |
| 126 | return best_guess |
| 127 | |
| 128 | |
| 129 | def classify_folder_images(folder_path, return_dict=False): |
no test coverage detected