AssertionError: The ./pretrain_models.pdparams does not exists! #13331

aqingGitHub · 2024-07-10T07:30:02Z

aqingGitHub
Jul 10, 2024

在开始训练输入 python tools/train.py -c configs/det/ch_ppocr_v2.0/ch_det_res18_db_v2.0.yml 后报错如下，请教下怎么解决？
(paddle2) D:\ProgramData\PaddleOCR-2.7.5>python tools/train.py -c configs/det/ch_ppocr_v2.0/ch_det_res18_db_v2.0.yml
[2024/07/10 14:01:11] ppocr INFO: Architecture :
[2024/07/10 14:01:11] ppocr INFO: Backbone :
[2024/07/10 14:01:11] ppocr INFO: disable_se : True
[2024/07/10 14:01:11] ppocr INFO: layers : 18
[2024/07/10 14:01:11] ppocr INFO: name : ResNet_vd
[2024/07/10 14:01:11] ppocr INFO: Head :
[2024/07/10 14:01:11] ppocr INFO: k : 50
[2024/07/10 14:01:11] ppocr INFO: name : DBHead
[2024/07/10 14:01:11] ppocr INFO: Neck :
[2024/07/10 14:01:11] ppocr INFO: name : DBFPN
[2024/07/10 14:01:11] ppocr INFO: out_channels : 256
[2024/07/10 14:01:11] ppocr INFO: Transform : None
[2024/07/10 14:01:11] ppocr INFO: algorithm : DB
[2024/07/10 14:01:11] ppocr INFO: model_type : det
[2024/07/10 14:01:11] ppocr INFO: Eval :
[2024/07/10 14:01:11] ppocr INFO: dataset :
[2024/07/10 14:01:11] ppocr INFO: data_dir : ./train_data/
[2024/07/10 14:01:11] ppocr INFO: label_file_list : ['./train_data/det/val.txt']
[2024/07/10 14:01:11] ppocr INFO: name : SimpleDataSet
[2024/07/10 14:01:11] ppocr INFO: transforms :
[2024/07/10 14:01:11] ppocr INFO: DecodeImage :
[2024/07/10 14:01:11] ppocr INFO: channel_first : False
[2024/07/10 14:01:11] ppocr INFO: img_mode : BGR
[2024/07/10 14:01:11] ppocr INFO: DetLabelEncode : None
[2024/07/10 14:01:11] ppocr INFO: DetResizeForTest : None
[2024/07/10 14:01:11] ppocr INFO: NormalizeImage :
[2024/07/10 14:01:11] ppocr INFO: mean : [0.485, 0.456, 0.406]
[2024/07/10 14:01:11] ppocr INFO: order : hwc
[2024/07/10 14:01:11] ppocr INFO: scale : 1./255.
[2024/07/10 14:01:11] ppocr INFO: std : [0.229, 0.224, 0.225]
[2024/07/10 14:01:11] ppocr INFO: ToCHWImage : None
[2024/07/10 14:01:11] ppocr INFO: KeepKeys :
[2024/07/10 14:01:11] ppocr INFO: keep_keys : ['image', 'shape', 'polys', 'ignore_tags']
[2024/07/10 14:01:11] ppocr INFO: loader :
[2024/07/10 14:01:11] ppocr INFO: batch_size_per_card : 1
[2024/07/10 14:01:11] ppocr INFO: drop_last : False
[2024/07/10 14:01:11] ppocr INFO: num_workers : 2
[2024/07/10 14:01:11] ppocr INFO: shuffle : False
[2024/07/10 14:01:11] ppocr INFO: Global :
[2024/07/10 14:01:11] ppocr INFO: cal_metric_during_train : False
[2024/07/10 14:01:11] ppocr INFO: checkpoints : None
[2024/07/10 14:01:11] ppocr INFO: distributed : False
[2024/07/10 14:01:11] ppocr INFO: epoch_num : 1200
[2024/07/10 14:01:11] ppocr INFO: eval_batch_step : [3000, 2000]
[2024/07/10 14:01:11] ppocr INFO: infer_img : doc/imgs_en/img_10.jpg
[2024/07/10 14:01:11] ppocr INFO: log_smooth_window : 20
[2024/07/10 14:01:11] ppocr INFO: pretrained_model : ./pretrain_models
[2024/07/10 14:01:11] ppocr INFO: print_batch_step : 2
[2024/07/10 14:01:11] ppocr INFO: save_epoch_step : 1200
[2024/07/10 14:01:11] ppocr INFO: save_inference_dir : None
[2024/07/10 14:01:11] ppocr INFO: save_model_dir : ./output/ch_db_res18/
[2024/07/10 14:01:11] ppocr INFO: save_res_path : ./output/det_db/predicts_db.txt
[2024/07/10 14:01:11] ppocr INFO: use_gpu : True
[2024/07/10 14:01:11] ppocr INFO: use_visualdl : False
[2024/07/10 14:01:11] ppocr INFO: Loss :
[2024/07/10 14:01:11] ppocr INFO: alpha : 5
[2024/07/10 14:01:11] ppocr INFO: balance_loss : True
[2024/07/10 14:01:11] ppocr INFO: beta : 10
[2024/07/10 14:01:11] ppocr INFO: main_loss_type : DiceLoss
[2024/07/10 14:01:11] ppocr INFO: name : DBLoss
[2024/07/10 14:01:11] ppocr INFO: ohem_ratio : 3
[2024/07/10 14:01:11] ppocr INFO: Metric :
[2024/07/10 14:01:11] ppocr INFO: main_indicator : hmean
[2024/07/10 14:01:11] ppocr INFO: name : DetMetric
[2024/07/10 14:01:11] ppocr INFO: Optimizer :
[2024/07/10 14:01:11] ppocr INFO: beta1 : 0.9
[2024/07/10 14:01:11] ppocr INFO: beta2 : 0.999
[2024/07/10 14:01:11] ppocr INFO: lr :
[2024/07/10 14:01:11] ppocr INFO: learning_rate : 0.001
[2024/07/10 14:01:11] ppocr INFO: name : Cosine
[2024/07/10 14:01:11] ppocr INFO: warmup_epoch : 2
[2024/07/10 14:01:11] ppocr INFO: name : Adam
[2024/07/10 14:01:11] ppocr INFO: regularizer :
[2024/07/10 14:01:11] ppocr INFO: factor : 0
[2024/07/10 14:01:11] ppocr INFO: name : L2
[2024/07/10 14:01:11] ppocr INFO: PostProcess :
[2024/07/10 14:01:11] ppocr INFO: box_thresh : 0.6
[2024/07/10 14:01:11] ppocr INFO: max_candidates : 1000
[2024/07/10 14:01:11] ppocr INFO: name : DBPostProcess
[2024/07/10 14:01:11] ppocr INFO: thresh : 0.3
[2024/07/10 14:01:11] ppocr INFO: unclip_ratio : 1.5
[2024/07/10 14:01:11] ppocr INFO: Train :
[2024/07/10 14:01:11] ppocr INFO: dataset :
[2024/07/10 14:01:11] ppocr INFO: data_dir : ./train_data/
[2024/07/10 14:01:11] ppocr INFO: label_file_list : ['./train_data/det/train.txt']
[2024/07/10 14:01:11] ppocr INFO: name : SimpleDataSet
[2024/07/10 14:01:11] ppocr INFO: ratio_list : [1.0]
[2024/07/10 14:01:11] ppocr INFO: transforms :
[2024/07/10 14:01:11] ppocr INFO: DecodeImage :
[2024/07/10 14:01:11] ppocr INFO: channel_first : False
[2024/07/10 14:01:11] ppocr INFO: img_mode : BGR
[2024/07/10 14:01:11] ppocr INFO: DetLabelEncode : None
[2024/07/10 14:01:11] ppocr INFO: IaaAugment :
[2024/07/10 14:01:11] ppocr INFO: augmenter_args :
[2024/07/10 14:01:11] ppocr INFO: args :
[2024/07/10 14:01:11] ppocr INFO: p : 0.5
[2024/07/10 14:01:11] ppocr INFO: type : Fliplr
[2024/07/10 14:01:11] ppocr INFO: args :
[2024/07/10 14:01:11] ppocr INFO: rotate : [-10, 10]
[2024/07/10 14:01:11] ppocr INFO: type : Affine
[2024/07/10 14:01:11] ppocr INFO: args :
[2024/07/10 14:01:11] ppocr INFO: size : [0.5, 3]
[2024/07/10 14:01:11] ppocr INFO: type : Resize
[2024/07/10 14:01:11] ppocr INFO: EastRandomCropData :
[2024/07/10 14:01:11] ppocr INFO: keep_ratio : True
[2024/07/10 14:01:11] ppocr INFO: max_tries : 50
[2024/07/10 14:01:11] ppocr INFO: size : [960, 960]
[2024/07/10 14:01:11] ppocr INFO: MakeBorderMap :
[2024/07/10 14:01:11] ppocr INFO: shrink_ratio : 0.4
[2024/07/10 14:01:11] ppocr INFO: thresh_max : 0.7
[2024/07/10 14:01:11] ppocr INFO: thresh_min : 0.3
[2024/07/10 14:01:11] ppocr INFO: MakeShrinkMap :
[2024/07/10 14:01:11] ppocr INFO: min_text_size : 8
[2024/07/10 14:01:11] ppocr INFO: shrink_ratio : 0.4
[2024/07/10 14:01:11] ppocr INFO: NormalizeImage :
[2024/07/10 14:01:11] ppocr INFO: mean : [0.485, 0.456, 0.406]
[2024/07/10 14:01:11] ppocr INFO: order : hwc
[2024/07/10 14:01:11] ppocr INFO: scale : 1./255.
[2024/07/10 14:01:11] ppocr INFO: std : [0.229, 0.224, 0.225]
[2024/07/10 14:01:11] ppocr INFO: ToCHWImage : None
[2024/07/10 14:01:11] ppocr INFO: KeepKeys :
[2024/07/10 14:01:11] ppocr INFO: keep_keys : ['image', 'threshold_map', 'threshold_mask', 'shrink_map', 'shrink_mask']
[2024/07/10 14:01:11] ppocr INFO: loader :
[2024/07/10 14:01:11] ppocr INFO: batch_size_per_card : 8
[2024/07/10 14:01:11] ppocr INFO: drop_last : False
[2024/07/10 14:01:11] ppocr INFO: num_workers : 4
[2024/07/10 14:01:11] ppocr INFO: shuffle : True
[2024/07/10 14:01:11] ppocr INFO: profiler_options : None
[2024/07/10 14:01:11] ppocr INFO: train with paddle 2.6.1 and device Place(gpu:0)
[2024/07/10 14:01:11] ppocr INFO: Initialize indexs of datasets:['./train_data/det/train.txt']
[2024/07/10 14:01:11] ppocr INFO: Initialize indexs of datasets:['./train_data/det/val.txt']
W0710 14:01:11.204589 22516 gpu_resources.cc:119] Please NOTE: device: 0, GPU Compute Capability: 6.1, Driver API Version: 11.8, Runtime API Version: 11.8
W0710 14:01:11.209575 22516 gpu_resources.cc:164] device: 0, cuDNN Version: 8.4.
[2024/07/10 14:01:12] ppocr INFO: train dataloader has 4 iters
[2024/07/10 14:01:12] ppocr INFO: valid dataloader has 8 iters
Traceback (most recent call last):
File "tools/train.py", line 229, in
main(config, device, logger, vdl_writer, seed)
File "tools/train.py", line 194, in main
pre_best_model_dict = load_model(config, model, optimizer,
File "D:\ProgramData\PaddleOCR-2.7.5\ppocr\utils\save_load.py", line 140, in load_model
is_float16 = load_pretrained_params(model, pretrained_model)
File "D:\ProgramData\PaddleOCR-2.7.5\ppocr\utils\save_load.py", line 152, in load_pretrained_params
assert os.path.exists(path + ".pdparams"),
AssertionError: The ./pretrain_models.pdparams does not exists!

Answered by aqingGitHub

Jul 12, 2024

.坑爹啊，安照网上的写法不行，要改成这样才可以，谢谢提醒！
.\pretrain_models\ch_ppocr_server_v2.0_det_train\best_accuracy

View full answer

UserWangZz · 2024-07-11T07:20:23Z

UserWangZz
Jul 11, 2024
Collaborator

pretrain_models/{预训练模型文件名（不带后缀）}
试一试

1 reply

aqingGitHub Jul 12, 2024
Author

.坑爹啊，安照网上的写法不行，要改成这样才可以，谢谢提醒！
.\pretrain_models\ch_ppocr_server_v2.0_det_train\best_accuracy

Answer selected by aqingGitHub

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

AssertionError: The ./pretrain_models.pdparams does not exists! #13331

{{title}}

{{editor}}'s edit

{{editor}}'s edit

Replies: 1 comment 1 reply

{{title}}

{{title}}

Select a reply

AssertionError: The ./pretrain_models.pdparams does not exists! #13331

aqingGitHub Jul 10, 2024

Replies: 1 comment · 1 reply

UserWangZz Jul 11, 2024 Collaborator

aqingGitHub Jul 12, 2024 Author

aqingGitHub
Jul 10, 2024

Replies: 1 comment 1 reply

UserWangZz
Jul 11, 2024
Collaborator

aqingGitHub Jul 12, 2024
Author