search

PaddlePaddle / PaddleOCR

Awesome multilingual OCR toolkits based on PaddlePaddle (practical ultra lightweight OCR system, support 80+ languages recognition, provide data annotation and synthesis tools, support training and deployment among server, mobile, embedded and IoT devices)
https://paddlepaddle.github.io/PaddleOCR/
Apache License 2.0
43.79k stars 7.79k forks source link

训练文字检测模型，在验证阶段报错 #1078

Closed simplew2011 closed 3 years ago

simplew2011 commented 3 years ago

将ccpd数据集的车牌区域作为文字检测数据集（100K），MobileNetV3_large_x0_5_pretrained作为预训练模型，训练报错。 每次出现在验证阶段。 日志和配置如下： train_det.log

试着修改以下三个地方，问题依旧。 return paddle.reader.multiprocess_reader(readers, False, queue_size=200) test_batch_size_per_card: 1 num_workers: 1

环境： Ubuntu16.04 GPU 2080 11G Cuda 10.0 cuDNN 7.6

Paddlepaddle-gpu 2.0.0b0 PaddleOCR v1.1

LDOUBLEV commented 3 years ago

image

你有改其他地方吗，报错提示这里，elementwise_add的两个输入shape不匹配，可以根据报错信息调试下

simplew2011 commented 3 years ago

image

你有改其他地方吗，报错提示这里，elementwise_add的两个输入shape不匹配，可以根据报错信息调试下

只修改过两个配置表，和reader_main.py中的： return paddle.reader.multiprocess_reader(readers, False, queue_size=200) 其他没修过

simplew2011 commented 3 years ago

订正： 修改了_reader.yml文件的test_image_shape,。因为ccpd数据集的图像尺寸是720*1160。 原来： EvalReader: test_image_shape: [736, 1280] 修改后： EvalReader: test_image_shape: [720, 1160]

图像尺寸无法32整除导致的报错么，可否在代码上判断，自动padding.

MissPenguin commented 3 years ago

订正： 修改了_reader.yml文件的test_image_shape,。因为ccpd数据集的图像尺寸是720*1160。 原来： EvalReader: test_image_shape: [736, 1280] 修改后： EvalReader: test_image_shape: [720, 1160]

图像尺寸无法32整除导致的报错么，可否在代码上判断，自动padding.

对的，需要是32的倍数。有需要的话可以自己试着改下代码。