Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
det_east_mv3.yml 1.01 KB
Newer Older
LDOUBLEV's avatar
upload PaddleOCR code  
LDOUBLEV committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43 
Global:
  algorithm: EAST
  use_gpu: true
  epoch_num: 100000
  log_smooth_window: 20
  print_batch_step: 5
  save_model_dir: output
  save_epoch_step: 200
  eval_batch_step: 5000
  train_batch_size_per_card: 16
  test_batch_size_per_card: 16
  image_shape: [3, 512, 512]
  reader_yml: ./configs/det/det_east_icdar15_reader.yml
  pretrain_weights: ./pretrain_models/MobileNetV3_pretrained/MobileNetV3_large_x0_5_pretrained/
  save_res_path: ./output/predicts_east.txt
  
Architecture:
  function: ppocr.modeling.architectures.det_model,DetModel

Backbone:
  function: ppocr.modeling.backbones.det_mobilenet_v3,MobileNetV3
  scale: 0.5
  model_name: large

Head:
  function: ppocr.modeling.heads.det_east_head,EASTHead
  model_name: small
  
Loss:
  function: ppocr.modeling.losses.det_east_loss,EASTLoss

Optimizer:
  function: ppocr.optimizer,AdamDecay
  base_lr: 0.001
  beta1: 0.9
  beta2: 0.999

PostProcess:
  function: ppocr.postprocess.east_postprocess,EASTPostPocess
  score_thresh: 0.8
  cover_thresh: 0.1
  nms_thresh: 0.2