txt_f = open(txt_file, 'w') FileNotFoundError: [Errno 2] No such file or directory: './test_result\\

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’)

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值