PaddleSeg icon indicating copy to clipboard operation
PaddleSeg copied to clipboard

[Bug] 自训练segformer算法模型,进行预测的时候加载不了模型参数

Open chenxi00a1 opened this issue 3 years ago • 1 comments

Thanks for your bug report. To help us solve the issue better, please provide following information:

  1. PaddleSeg version: (PaddleSeg release/2.3)

  2. PaddlePaddle version: (paddlepaddle-gpu 2.3.0.post112)

  3. Operation system: (Linux)

  4. Python version: (Python3.6.13)

  5. CUDA/cuDNN version: (CUDA11.2/cuDNN 8.1)

  6. Full codes: train : python train.py --config configs\segformer\segformer_b4_cityscapes_1024x1024_160k.yml --do_eval --use_vdl --save_interval 500 --save_dir output predict: python predict.py --config configs/segformer/segformer_b4_cityscapes_1024x1024_160k.yml --model_path output/best_model/model.pdparams --image_path data/data_ys/JPEGImages/1654073382157737_c6_98.png --save_dir output/result

  7. Detailed error information, releated running log: 可以看到,模型加载路径没问题,也没加载错误模型(比如训练的是Segfomer 而加载的PP_Lite) 在加载(0/847)参数后,直接开始预测,预测结果很花(估计是没有加载模型参数,模型参数是一些随机数) 1657788493353

另外:训练segfomer-b4 是正常的(虽然训练效果不佳,由于数据问题)

2

chenxi00a1 avatar Jul 14 '22 08:07 chenxi00a1

考虑output中模型参数是否不对呢?日志提醒模型参数没有加载上

shiyutang avatar Jul 15 '22 14:07 shiyutang