This commit is contained in:
Kabooshki 2024-09-17 13:04:49 -05:00
parent cf8af5ab22
commit 1040e754e5
Signed by: Kabooshki
GPG Key ID: 5E7B7F2F405CD62A
1 changed files with 7 additions and 19 deletions

24
main.py
View File

@ -1,9 +1,9 @@
import cv2
import pytesseract
import os
import numpy
directory = 'images'
k = 1
# iterate over files in
# that directory
for filename in os.listdir(directory):
@ -20,21 +20,9 @@ for filename in os.listdir(directory):
print(data)
if os.path.isfile(f'images/{data}.jpg') == False:
os.rename(f'{f}', f'images/{data}.jpg')
elif os.path.isfile(f'images/{data}.jpg') == True:
cv2.imshow(data, ROI)
cv2.waitKey(1)
manpage = input("please input the number on the page (if the title is correct, enter nothing)")
print(manpage)
if manpage == "\n":
rng = numpy.random.default_rng().random()
os.rename (f'images/{data}.jpg', f'images/{data}.review-{rng}.jpg')
os.rename(f'{f}', f'images/{data}.jpg')
elif manpage != "\n":
if os.path.isfile(f'images/{manpage}.jpg') == True:
rng = numpy.random.default_rng().random()
os.rename (f'images/{manpage}.jpg', f'images/{manpage}.review-{rng}.jpg')
os.rename(f'{f}', f'images/{manpage}.jpg')
elif os.path.isfile(f'images/{manpage}.jpg') == False:
os.rename(f'{f}', f'images/{manpage}.jpg')
cv2.destroyAllWindows()
else:
cv2.imshow(ROI)
cv2.waitKey(50)
os.rename(f'{f}', f'images/{data}.{k}.jpg')
k = k + 1
cv2.destroyAllWindows()