OWN_config.yaml 904 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. GPUID: 0
  2. WORKERS: 1
  3. PRINT_FREQ: 10
  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: OWN
  13. ROOT: "H:/DL-DATASET/360M/images"
  14. JSON_FILE: {'train': 'lib/dataset/txt/train_own.txt', 'val': 'lib/dataset/txt/test_own.txt'}
  15. SCALE_FACTOR: 0.25
  16. ROT_FACTOR: 30
  17. STD: 0.193
  18. MEAN: 0.588
  19. ALPHABETS: ''
  20. TRAIN:
  21. BATCH_SIZE_PER_GPU: 32
  22. SHUFFLE: True
  23. BEGIN_EPOCH: 0
  24. END_EPOCH: 100
  25. RESUME:
  26. IS_RESUME: False
  27. FILE: ''
  28. OPTIMIZER: 'adam'
  29. LR: 0.0001
  30. WD: 0.0
  31. LR_STEP: [60, 80]
  32. LR_FACTOR: 0.1
  33. MOMENTUM: 0.0
  34. NESTEROV: False
  35. RMSPROP_ALPHA:
  36. RMSPROP_CENTERED:
  37. TEST:
  38. BATCH_SIZE_PER_GPU: 16
  39. SHUFFLE: True
  40. NUM_TEST: 1000
  41. NUM_TEST_DISP: 10
  42. MODEL:
  43. NAME: 'crnn'
  44. IMAGE_SIZE:
  45. OW: 280. # origial width: 280
  46. H: 32
  47. W: 160 # resized width: 160
  48. NUM_CLASSES: 0
  49. NUM_HIDDEN: 256