import os
from ocr import ocr
import time
import shutil
import numpy as np
from PIL import Image
from glob import glob
def single_pic_proc(image_file):
image = np.array(Image.open(image_file).convert(‘RGB’))
result, image_framed = ocr(image)
return result,image_framed
if name == ‘main’:
image_files = glob(’./test_images/.’)
result_dir = ‘./test_result’
if os.path.exists(result_dir):
shutil.rmtree(result_dir)
os.mkdir(result_dir) # 此处应改为# os.makedirs(’./test_result/test_images’)
for image_file in sorted(image_files):
t = time.time()
result, image_framed = single_pic_proc(image_file)
output_file = os.path.join(result_dir, image_file.split('/')[-1])
txt_file = os.path.join(result_dir, image_file.split('/')[-1].split('.')[0]+'.txt')
print(txt_file)
txt_f = open(txt_file, 'w')
Image.fromarray(image_framed).save(output_file)
print("Mission complete, it took {:.3f}s".format(time.time() - t))
print("\nRecognition Result:\n")
for key in result:
print(result[key][1])
txt_f.write(result[key][1]+'\n')
txt_f.close()
执行报错:Traceback (most recent call last):
File “E:/Python/Pycharm/my work/AI_study/Pytorch_india_learning/Pytorch_learning/Tangyudi_video/chapter11 OCR/ocr.pytorch-master/demo.py”, line 31, in
txt_f = open(txt_file, ‘w’)
FileNotFoundError: [Errno 2] No such file or directory: ‘./test_result\test_images\t1.txt’
os.mkdir(result_dir) 此处应改为 os.makedirs(’./test_result/test_images’)