search

YCG09 / chinese_ocr

CTPN + DenseNet + CTC based end-to-end Chinese OCR implemented using tensorflow and keras
Apache License 2.0
2.73k stars 1.08k forks source link

无法正常执行demo.py,求指点 #364

Closed lmw0320 closed 3 years ago

lmw0320 commented 3 years ago

我在win10下跑demo.py文件时，发现无法正常检测。因此我加了断点打印，想查看哪里出了问题，结果发现只能执行到创建文件处，后面就直接跳出了，很是奇怪。。

1 2

lmw0320 commented 3 years ago

其中还出了个问题： Loading network VGGnet_test... Restoring from d:\linminwei\python_files\CV\OCR\YCG09-ocr\ctpn/checkpoints\VGGnet_fast_rcnn_iter_50000.ckpt... 2020-08-25 17:49:00.265894: I tensorflow/stream_executor/cuda/cuda_driver.cc:831] failed to allocate 8.00G (8589934592 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory 2020-08-25 17:49:00.266856: I tensorflow/stream_executor/cuda/cuda_driver.cc:831] failed to allocate 7.20G (7730940928 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory

我很奇怪，为什么中间会报这个OOM？？8G的显存，不至于检测一张图片就爆了吧？？而且后面又是正常继续执行下去的。。

lmw0320 commented 3 years ago

后面查出来了，是我自己的问题，image_files读取为空，需要设置绝对路径才行。。