PP-OCRv5_mobile_det.yaml 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. Global:
  2. model: PP-OCRv5_mobile_det
  3. mode: check_dataset # check_dataset/train/evaluate/predict
  4. module: text_det
  5. dataset_dir: "/paddle/dataset/paddlex/ocr_det/ocr_det_dataset_examples"
  6. device: gpu:0,1,2,3
  7. output: "output"
  8. CheckDataset:
  9. convert:
  10. enable: False
  11. src_dataset_type: null
  12. split:
  13. enable: False
  14. train_percent: null
  15. val_percent: null
  16. Train:
  17. epochs_iters: 100
  18. batch_size: 4
  19. learning_rate: 0.001
  20. pretrain_weight_path: "PP-OCRv5_mobile_det_pretrained.pdparams"
  21. resume_path: null
  22. log_interval: 10
  23. eval_interval: 1
  24. save_interval: 1
  25. Evaluate:
  26. weight_path: "output/best_accuracy/best_accuracy.pdparams"
  27. log_interval: 1
  28. Export:
  29. weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-OCRv5_mobile_det_pretrained.pdparams
  30. Predict:
  31. batch_size: 1
  32. model_dir: "output/best_accuracy/inference"
  33. input: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_001.png"
  34. kernel_option:
  35. run_mode: paddle