时间:2024年12月4日
遇到的困难
1、-c或者-o出现一次就可以
python tools/train.py \
-c data/en_PP-OCRv3_rec.yml \
-o Global.pretrained_model=./pretrain_models/en_PP-OCRv3_rec_train/best_accuracy \
Global.epoch_num=1 \
Train.loader.batch_size_per_card=16
2、检测推理报错FatalError: Segmentation fault is detected by the operating system.
当时从网上面查和版本由关系。更换版本后还是不行。最后将检测模型转变为推理模型,然后实现推理
3、初始化环境变量
#!/bin/bash
# python.
export CUDA_VISIBLE_DEVICES=1
export PYTHONHOME=$PWD/code/py38_ocr
export PATH=$PYTHONHOME/bin:$PATH
export LD_LIBRARY_PATH=$PYTHONHOME/lib:$LD_LIBRARY_PATH
export PYTHONUNBUFFERED=1
其中这个文件可以获取到当前所在文件的目录
$PWD
4、预训练模型加载的问题
使用官方推荐的检测预训练模型,训练速度会很慢,这个预训练模型文件可能是提取图片特征,并不具备OCR检测,所以直接下载v3或者v4版本的模型及其配置文件就可以