PaddleOCR export_model.py : TypeError: can only concatenate list (not "NoneType") to list

 

PaddleOCR recognition model 학습을 완료했다.
tools/eval.py 로 테스트를 진행해 보았고, gpu를 사용하여 inference 결과도 보고 싶어서 모델을 export_model 을 적용하여  inference 용 모델로 전환하려했다.

 

python3 tools/export_model.py -c /workspace/PaddleOCR/output/rec_ppocr_v3_korean_v4/config.yml -o Global.pretrained_model=./output/rec_ppocr_v3_korean_v4/best_accuracy Global.save_inference_dir=./inference/korean_PP-OCRv3_rec/

W0610 10:30:36.766150 1797163 gpu_resources.cc:119] Please NOTE: device: 0, GPU Compute Capability: 8.9, Driver API Version: 12.2, Runtime API Version: 12.0
W0610 10:30:36.767215 1797163 gpu_resources.cc:164] device: 0, cuDNN Version: 8.9.
[2024/06/10 10:30:36] ppocr INFO: load pretrain successful from ./output/rec_ppocr_v3_korean_v4/best_accuracy
Traceback (most recent call last):
  File "/workspace/PaddleOCR/tools/export_model.py", line 295, in <module>
    main()
  File "/workspace/PaddleOCR/tools/export_model.py", line 289, in main
    export_single_model(
  File "/workspace/PaddleOCR/tools/export_model.py", line 68, in export_single_model
    paddle.static.InputSpec(shape=[None] + input_shape, dtype="float32"),
TypeError: can only concatenate list (not "NoneType") to list

 

 

 

해결 방법

해결 방법은 paddleOCR release 2.6 버전에 있는 export_model.py  를 복사 붙여넣기해서 현재 paddleOCR/tolls/export_model.py 에 붙여넣기 하면 된다.

https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.6/tools/export_model.py

PaddleOCR/tools/export_model.py at release/2.6 · PaddlePaddle/PaddleOCR

 

이렇게 하면 inference 용 모델이 생성된 것을 볼 수있다.