wangwei
/
dface
Not watched
1
0
Fork 0
Code
Releases 2 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 9a0ce7b979
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '9a0ce7b979'
${ noResults }
dface/src/config.py

42 lines
1.0 kB
Raw Blame History 

  1. import os

  2. 

  3. 

  4. MODEL_STORE_DIR = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))+"/model_store"

  5. 

  6. 

  7. ANNO_STORE_DIR = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))+"/anno_store"

  8. 

  9. 

  10. LOG_DIR = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))+"/log"

  11. 

  12. 

  13. USE_CUDA = True

  14. 

  15. 

  16. TRAIN_BATCH_SIZE = 512

  17. 

  18. TRAIN_LR = 0.01

  19. 

  20. END_EPOCH = 10

  21. 

  22. 

  23. PNET_POSTIVE_ANNO_FILENAME = "pos_12.txt"

  24. PNET_NEGATIVE_ANNO_FILENAME = "neg_12.txt"

  25. PNET_PART_ANNO_FILENAME = "part_12.txt"

  26. PNET_LANDMARK_ANNO_FILENAME = "landmark_12.txt"

  27. 

  28. 

  29. RNET_POSTIVE_ANNO_FILENAME = "pos_24.txt"

  30. RNET_NEGATIVE_ANNO_FILENAME = "neg_24.txt"

  31. RNET_PART_ANNO_FILENAME = "part_24.txt"

  32. RNET_LANDMARK_ANNO_FILENAME = "landmark_24.txt"

  33. 

  34. 

  35. ONET_POSTIVE_ANNO_FILENAME = "pos_48.txt"

  36. ONET_NEGATIVE_ANNO_FILENAME = "neg_48.txt"

  37. ONET_PART_ANNO_FILENAME = "part_48.txt"

  38. ONET_LANDMARK_ANNO_FILENAME = "landmark_48.txt"

  39. 

  40. PNET_TRAIN_IMGLIST_FILENAME = "imglist_anno_12.txt"

  41. RNET_TRAIN_IMGLIST_FILENAME = "imglist_anno_24.txt"

  42. ONET_TRAIN_IMGLIST_FILENAME = "imglist_anno_48.txt"