config.yaml 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. # Project definitions (do not edit)
  2. Task: eyes
  3. scorer: Keisuke
  4. date: Dec8
  5. # Project path (change when moving around)
  6. project_path: /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eyes-Keisuke-2019-12-08
  7. # Annotation data set configuration (and individual video cropping parameters)
  8. video_sets:
  9. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA007518_20191006_3-0.avi:
  10. crop: 0, 192, 0, 200
  11. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA007518_20191006_3-1.avi:
  12. crop: 0, 192, 0, 200
  13. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA005759_20190324_144801-0.avi:
  14. crop: 0, 192, 0, 200
  15. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA005759_20190324_144801-1.avi:
  16. crop: 0, 192, 0, 200
  17. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA005719_20190929_3-0.avi:
  18. crop: 0, 192, 0, 200
  19. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA005719_20190929_3-1.avi:
  20. crop: 0, 192, 0, 200
  21. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA005756_20190625_5-0.avi:
  22. crop: 0, 192, 0, 200
  23. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA005756_20190625_5-1.avi:
  24. crop: 0, 192, 0, 200
  25. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA007296_20190929_5-0.avi:
  26. crop: 0, 192, 0, 200
  27. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA007296_20190929_5-1.avi:
  28. crop: 0, 192, 0, 200
  29. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA005758_20190322_145446-0.avi:
  30. crop: 0, 192, 0, 200
  31. /media/mouse/Data2/AirTrack_PupilTrack/EyeTrackingModel/eye_videos/MLA005758_20190322_145446-1.avi:
  32. crop: 0, 192, 0, 200
  33. bodyparts:
  34. - eye_top
  35. - eye_topleft
  36. - eye_left
  37. - eye_bottomleft
  38. - eye_bottom
  39. - eye_bottomright
  40. - eye_right
  41. - eye_topright
  42. - pupil_top
  43. - pupil_topleft
  44. - pupil_left
  45. - pupil_bottomleft
  46. - pupil_bottom
  47. - pupil_bottomright
  48. - pupil_right
  49. - pupil_topright
  50. start: 0
  51. stop: 1
  52. numframes2pick: 12
  53. # Plotting configuration
  54. skeleton:
  55. - - pupil_top
  56. - pupil_topleft
  57. - - pupil_topleft
  58. - pupil_left
  59. - - pupil_left
  60. - pupil_bottomleft
  61. - - pupil_bottomleft
  62. - pupil_bottom
  63. - - pupil_bottom
  64. - pupil_bottomright
  65. - - pupil_bottomright
  66. - pupil_right
  67. - - pupil_right
  68. - pupil_topright
  69. - - pupil_topright
  70. - pupil_top
  71. skeleton_color: yellow
  72. pcutoff: 0.1
  73. dotsize: 4
  74. alphavalue: 0.5
  75. colormap: gist_rainbow
  76. # Training,Evaluation and Analysis configuration
  77. TrainingFraction:
  78. - 0.95
  79. iteration: 6
  80. resnet:
  81. snapshotindex: -1
  82. batch_size: 8
  83. # Cropping Parameters (for analysis and outlier frame detection)
  84. cropping: false
  85. #if cropping is true for analysis, then set the values here:
  86. x1: 0
  87. x2: 640
  88. y1: 277
  89. y2: 624
  90. # Refinement configuration (parameters from annotation dataset configuration also relevant in this stage)
  91. corner2move2:
  92. - 50
  93. - 50
  94. move2corner: true
  95. default_net_type: resnet_50
  96. default_augmenter: default