paddleocr训练文本检测模型，跑1个epoch后自动退出

[wangpeng198754]

Global: use_gpu: true epoch_num: 1200 log_smooth_window: 20 print_batch_step: 10 save_model_dir: ./output/db_mv3/ save_epoch_step: 3

evaluation is run every 2000 iterations

eval_batch_step: [0,400]

1. If pretrained_model is saved in static mode, such as classification pretrained model

from static branch, load_static_weights must be set as True.

2. If you want to finetune the pretrained models we provide in the docs,

you should set load_static_weights as False.

load_static_weights: True cal_metric_during_train: False pretrained_model: ./pretrain_models/MobileNetV3_large_x0_5_pretrained/ checkpoints: save_inference_dir: ./output/det_db/ use_visualdl: False infer_img: ./doc/imgs_en/img_10.jpg save_res_path: ./output/det_db/predicts_db.txt

Architecture: model_type: det algorithm: DB Transform: Backbone: name: MobileNetV3 scale: 0.5 model_name: large Neck: name: DBFPN out_channels: 256 Head: name: DBHead k: 50

Loss: name: DBLoss balance_loss: true main_loss_type: DiceLoss alpha: 5 beta: 10 ohem_ratio: 3

Optimizer: name: Adam beta1: 0.9 beta2: 0.999 lr: learning_rate: 0.001 regularizer: name: 'L2' factor: 0

PostProcess: name: DBPostProcess thresh: 0.3 box_thresh: 0.6 max_candidates: 1000 unclip_ratio: 1.5

Metric: name: DetMetric main_indicator: hmean

Train: dataset: name: SimpleDataSet data_dir: C:/Users/Administrator/Desktop/tianchi/ label_file_list:

• C:/Users/Administrator/Desktop/tianchi/train.txt ratio_list: [1.0] transforms:
  • DecodeImage: # load image img_mode: BGR channel_first: False
  • DetLabelEncode: # Class handling label
  • IaaAugment: augmenter_args:
    • { 'type': Fliplr, 'args': { 'p': 0.5 } }
    • { 'type': Affine, 'args': { 'rotate': [-10, 10] } }
    • { 'type': Resize, 'args': { 'size': [0.5, 3] } }
  • EastRandomCropData: size: [640, 640] max_tries: 50 keep_ratio: true
  • MakeBorderMap: shrink_ratio: 0.4 thresh_min: 0.3 thresh_max: 0.7
  • MakeShrinkMap: shrink_ratio: 0.4 min_text_size: 8
  • NormalizeImage: scale: 1./255. mean: [0.485, 0.456, 0.406] std: [0.229, 0.224, 0.225] order: 'hwc'
  • ToCHWImage:
  • KeepKeys: keep_keys: ['image', 'threshold_map', 'threshold_mask', 'shrink_map', 'shrink_mask'] # the order of the dataloader list loader: shuffle: True drop_last: False batch_size_per_card: 4 num_workers: 2 use_shared_memory: False

Eval: dataset: name: SimpleDataSet data_dir: C:/Users/Administrator/Desktop/tianchi/ label_file_list:

• C:/Users/Administrator/Desktop/tianchi/test.txt transforms:
  • DecodeImage: # load image img_mode: BGR channel_first: False
  • DetLabelEncode: # Class handling label
  • DetResizeForTest: image_shape: [736, 1280]
  • NormalizeImage: scale: 1./255. mean: [0.485, 0.456, 0.406] std: [0.229, 0.224, 0.225] order: 'hwc'
  • ToCHWImage:
  • KeepKeys: keep_keys: ['image', 'shape', 'polys', 'ignore_tags'] loader: shuffle: False drop_last: False batch_size_per_card: 1 # must be 1 num_workers: 2 use_shared_memory: False

[tink2123]

可以把 num worker 修改为 0 试试

[Evezerest]

The issue will be closed, if it is not solved or there is a follow-up one, please reopen it at any time and we will continue to follow up. It is recommended to pull and try the latest code first. 该问题将会被关闭，若问题未解决或有后续问题，请随时重新打开（建议先拉取最新代码进行尝试），我们会继续跟进。