andwati/image-sort
1
0
Fork 0
mirror of https://github.com/andwati/image-sort.git synced 2026-10-23 18:32:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: added multithreading

Browse Source
This commit is contained in:
andwati
2023-06-16 10:25:54 +03:00
parent 4d87850d0e
commit b3b5b94dac
4 changed files with 25 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cli.py
+11 -8
View File
@@ -2,6 +2,7 @@ import argparse
from pathlib import Path
import cv2
from sorter.image_sorter import process_image
from concurrent.futures import ThreadPoolExecutor
# CLI argument parsing
def parse_arguments():
@@ -35,14 +36,16 @@ def main():
image_paths = list(input_folder.glob("*.jpg")) + list(input_folder.glob("*.png"))
# Process the images
for image_path in image_paths:
"""
try catch block to handle errors in processing individual images.
"""
try:
process_image(image_path, pass_folder, reject_folder)
except cv2.error as e:
print(f"Error processing image {image_path.name}: {str(e)}")
with ThreadPoolExecutor() as executor:
for image_path in image_paths:
"""
try catch block to handle errors in processing individual images.
"""
try:
#process_image(image_path, pass_folder, reject_folder)
executor.submit(process_image, image_path, pass_folder, reject_folder)
except cv2.error as e:
print(f"Error processing image {image_path.name}: {str(e)}")
print("Sorting complete.")