360CC_config.yaml 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. GPUID: 0
  2. WORKERS: 1
  3. PRINT_FREQ: 100
  4. SAVE_FREQ: 10
  5. PIN_MEMORY: False
  6. OUTPUT_DIR: 'output'
  7. CUDNN:
  8. BENCHMARK: True
  9. DETERMINISTIC: False
  10. ENABLED: True
  11. DATASET:
  12. DATASET: 360CC
  13. # ROOT: "H:/DL-DATASET/360M/images"
  14. ROOT: "C:/Users/Administrator/Desktop/OCR_pytorch/Attention_ocr.pytorch/test_img"
  15. CHAR_FILE: 'lib/dataset/txt/char_std_5990.txt'
  16. JSON_FILE: {'train': 'lib/dataset/txt/my_test.txt', 'val': 'lib/dataset/txt/test.txt'}
  17. # JSON_FILE: {'train': 'H:/DL-DATASET/360M/train.txt', 'val': 'H:/DL-DATASET/360M/test.txt'}
  18. SCALE_FACTOR: 0.25
  19. ROT_FACTOR: 30
  20. STD: 0.193
  21. MEAN: 0.588
  22. ALPHABETS: ''
  23. TRAIN:
  24. BATCH_SIZE_PER_GPU: 64
  25. SHUFFLE: True
  26. BEGIN_EPOCH: 0
  27. END_EPOCH: 100
  28. RESUME:
  29. IS_RESUME: False
  30. FILE: ''
  31. OPTIMIZER: 'adam'
  32. LR: 0.0001
  33. WD: 0.0
  34. LR_STEP: [60, 80]
  35. LR_FACTOR: 0.1
  36. MOMENTUM: 0.0
  37. NESTEROV: False
  38. RMSPROP_ALPHA:
  39. RMSPROP_CENTERED:
  40. FINETUNE:
  41. IS_FINETUNE: true
  42. FINETUNE_CHECKPOINIT: 'output/checkpoints/mixed_second_finetune_acc_97P7.pth'
  43. FREEZE: true
  44. TEST:
  45. BATCH_SIZE_PER_GPU: 16
  46. SHUFFLE: True # for random test rather than test on the whole validation set
  47. NUM_TEST_BATCH: 1000
  48. NUM_TEST_DISP: 10
  49. MODEL:
  50. NAME: 'crnn'
  51. IMAGE_SIZE:
  52. OW: 280 # origial width: 280
  53. H: 32
  54. W: 960 # resized width: 160
  55. NUM_CLASSES: 0
  56. NUM_HIDDEN: 256