PaddleX
PaddleX copied to clipboard
使用paddlex训练ppocrv5的文本检测模型速度很慢
我在使用paddlex进行ppocrv5的文本检测模型时,模型的训练过程很快,但是在每一个epoch完成后,进行eval很慢,一张图片需要接近5秒钟,我目前是一个测试的小规模数据集,val集只有11张图片,但是要等很久,在eval的过程中,GPU占用为0,CPU占用也很低,下面这张图是训练时GPU占用情况,验证集一张图可能有大几十到100多个bbox不等
能发一下你的训练配置文件吗?另外,你的验证集中的图像分辨是不是很大呀
能发一下你的训练配置文件吗?另外,你的验证集中的图像分辨是不是很大呀
你好,我的训练脚本:
python main.py -c paddlex/configs/modules/text_detection/PP-OCRv5_mobile_det.yaml ^
-o Global.output=F:/unattended_laboratory/OCR/det/checkpoint ^
-o Global.device=gpu:0 ^
-o Global.mode=train ^
-o Global.dataset_dir=F:/unattended_laboratory/OCR/det/dataset
我的验证集中的图片是1920x1080分辨率的,就是电脑的截屏然后检测屏幕上的文字。