Classify all images from a local folder.
(folder_path, return_dict=False)
| 127 | |
| 128 | |
| 129 | def classify_folder_images(folder_path, return_dict=False): |
| 130 | """ Classify all images from a local folder. """ |
| 131 | classified_images_list = [] |
| 132 | classified_images_dict = {} |
| 133 | files = [f for f in listdir(folder_path) if isfile(join(folder_path, f))] |
| 134 | images = [f for f in files if (f.endswith('.jpg') or f.endswith('.png'))] |
| 135 | total = len(images) |
| 136 | counter = 0 |
| 137 | for image in images: |
| 138 | counter += 1 |
| 139 | sys.stdout.write("\rClassifying Image %d of %s..." % (counter, total)) |
| 140 | sys.stdout.flush() |
| 141 | result = classify_local_image(os.path.join(folder_path, image)) |
| 142 | classified_images_list.append(result) |
| 143 | classified_images_dict[image] = result |
| 144 | sys.stdout.write("\rAll classifications have been completed!\n") |
| 145 | if return_dict: |
| 146 | return classified_images_dict |
| 147 | return classified_images_list |
| 148 | |
| 149 | |
| 150 | def classify(image_url_or_path): |
no test coverage detected