demo.py测试图片时，每张图片都很耗时，怎么解决。。。

lmw0320 commented 3 years ago

我看demo的预测代码中，对每张图片都进行遍历，loadmodel来检测，这样每次都很耗时。。。不知道有什么好办法解决么。。

lmw0320 commented 3 years ago

我查看了下，GPU占用很大（即使只有一张图片进行检测）。同时我为了确保GPU的使用，特地增加了如下代码，结果却无法正常检测： if tf.test.is_gpu_available(): #如果GPU可用，则指定一块GPU加速 os.environ["CUDA_VISIBLE_DEVICES"] = "0" #指定第一块GPU可用 config = tf.compat.v1.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 程序最多只能占用指定gpu90%的显存 config.gpu_options.allow_growth = True #不全部占满显存, 按需分配 sess = tf.compat.v1.Session(config=config) K.set_session(sess) 搞不懂几点：

为什么一张张图片进行检测，也会占用如此大的GPU资源？
既然使用了GPU资源，为什么每张图片的检测还是如此耗时？

lmw0320 commented 3 years ago

@YCG09

Cocoalate commented 3 years ago

同问啊，测了一下是K.get_value(K.ctc_decode ...）这个地方一张图片可能都需要1秒左右了

bai-0829 commented 6 months ago

有没有大佬能解答一下呀，测试一张图片，现存占满了，直接就卡在这里不动了

suoixnazx commented 2 months ago

有没有大佬能解答一下呀，测试一张图片，现存占满了，直接就卡在这里不动了

请问你解决了吗？

YCG09 / chinese_ocr

demo.py测试图片时，每张图片都很耗时，怎么解决。。。 #366