首页 PaddleOCR 帖子详情
PaddleOCR 加载预训练模型问题
收藏
快速回复
PaddleOCR 问答学习资料 1447 5
PaddleOCR 加载预训练模型问题
收藏
快速回复
PaddleOCR 问答学习资料 1447 5

PaddleOCR 加载预训练模型老出错

# 训练 mv3_db 模型,并将训练日志保存为 tain_det.log
%cd PaddleOCR/
!python3 tools/train.py -c configs/det/det_r50_vd_db.yml \
     -o Global.pretrain_weights=./pretrain_models/ResNet50_vd_ssld_pretrained \
     2>&1 | tee train_det.log

报错信息:

W1223 01:25:11.035835  4034 device_context.cc:447] Please NOTE: device: 0, GPU Compute Capability: 7.0, Driver API Version: 10.1, Runtime API Version: 10.1
W1223 01:25:11.040609  4034 device_context.cc:465] device: 0, cuDNN Version: 7.6.
Traceback (most recent call last):
  File "tools/train.py", line 148, in 
    main(config, device, logger, vdl_writer)
  File "tools/train.py", line 100, in main
    pre_best_model_dict = load_model(config, model, optimizer)
  File "/home/aistudio/PaddleOCR/ppocr/utils/save_load.py", line 98, in load_model
    load_pretrained_params(model, pretrained_model)
  File "/home/aistudio/PaddleOCR/ppocr/utils/save_load.py", line 109, in load_pretrained_params
    "The {}.pdparams does not exists!".format(path)
AssertionError: The ./pretrain_models/ResNet50_vd_ssld_pretrained.pdparams does not exists!

 

就是说默认预训练模型都是*.pdparams 格式,对于散列模型,怎么读取?

 

 

 

 

 

0
收藏
回复
全部评论(5)
时间顺序
JavaRoom
#2 回复于2021-12

用pdparams格式的即可

!wget -P ./pretrain_models/ https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ResNet50_vd_ssld_pretrained.pdparams
0
回复
stivenssss
#3 回复于2021-12

自动补齐后缀名“.pdparams”,因此最后一个是模型名

0
回复
JavaRoom
#4 回复于2021-12
自动补齐后缀名“.pdparams”,因此最后一个是模型名

对于那种一堆文件的模型怎么弄?

0
回复
stivenssss
#5 回复于2021-12
对于那种一堆文件的模型怎么弄?

你用的应该是静态版,动态版的就不会有那么一堆的

0
回复
stivenssss
#6 回复于2021-12
对于那种一堆文件的模型怎么弄?

静态版一堆的,是传所在文件夹路径就可以的了

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户