Home Explore Help
Register Sign In
Fengyang
/
conveying_belt_recognition
1
0
Fork 1
Files Issues 0 Pull Requests 0 Wiki
Branch: master
Branches Tags
conveying_be...
/
deep-hough-transform-master
/
config.yml

config.yml 1004 B
Permalink History Raw

12345678910111213141516171819202122232425262728293031323334353637383940414243 
  1. DATA:
    2. 

  2.     DIR: "data/training/"
    3. 

  3.     VAL_DIR: "data/training/"
    4. 

  4.     TEST_DIR: "data/NKL"
    5. 

  5.     LABEL_FILE: "data/training/nkl_train.txt"
    6. 

  6.     VAL_LABEL_FILE: "data/training/nkl_val.txt"
    7. 

  7.     TEST_LABEL_FILE: "data/training/nkl_test.txt"
    8. 

  8.     # DIR: "data/training/"
    9. 

  9.     # VAL_DIR: "data/training/"
    10. 

  10.     # TEST_DIR: "data/training"
    11. 

  11.     # LABEL_FILE: "data/training/train_1716_100_100.txt"
    12. 

  12.     # VAL_LABEL_FILE: "data/training/test_1716_100_100.txt"
    13. 

  13.     # TEST_LABEL_FILE: "data/training/test_1716_100_100.txt"
    14. 

  14.     BATCH_SIZE: 8
    15. 

  15.     WORKERS: 4
    16. 

  16. 

  17. OPTIMIZER:
    18. 

  18.     LR: 2e-4
    19. 

  19.     MOMENTUM: 0.9
    20. 

  20.     GAMMA: 0.1
    21. 

  21.     WEIGHT_DECAY: 0
    22. 

  22.     STEPS: []
    23. 

  23. 

  24. MODEL:
    25. 

  25.     NUMANGLE: 100
    26. 

  26.     NUMRHO: 100
    27. 

  27.     FIX: True
    28. 

  28.     THRESHOLD: 0.01
    29. 

  29.     EDGE_ALIGN: False
    30. 

  30.     BACKBONE: resnet50 
    31. 

  31.     # only support : resnet18, resnet50, resnet101, mobilenetv2, resnext50, res2net50, vgg16
    32. 

  32. 

  33. TRAIN:
    34. 

  34.     EPOCHS: 30
    35. 

  35.     PRINT_FREQ: 100
    36. 

  36.     TEST: False
    37. 

  37.     SEED: 1997
    38. 

  38.     GPU_ID: 0
    39. 

  39.     DATA_PARALLEL: False
    40. 

  40.     RESUME: 
    41. 

  41. 

  42. MISC:
    43. 

  43.     TMP: "./result/reproduce"
    44.