wflw.py 9.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192
  1. dataset_info = dict(
  2. dataset_name='wflw',
  3. paper_info=dict(
  4. author='Wu, Wayne and Qian, Chen and Yang, Shuo and Wang, '
  5. 'Quan and Cai, Yici and Zhou, Qiang',
  6. title='Look at boundary: A boundary-aware face alignment algorithm',
  7. container='Proceedings of the IEEE conference on computer '
  8. 'vision and pattern recognition',
  9. year='2018',
  10. homepage='https://wywu.github.io/projects/LAB/WFLW.html',
  11. ),
  12. keypoint_info={
  13. 0: dict(name='kpt-0', id=0, color=[255, 0, 0], type='', swap='kpt-32'),
  14. 1: dict(name='kpt-1', id=1, color=[255, 0, 0], type='', swap='kpt-31'),
  15. 2: dict(name='kpt-2', id=2, color=[255, 0, 0], type='', swap='kpt-30'),
  16. 3: dict(name='kpt-3', id=3, color=[255, 0, 0], type='', swap='kpt-29'),
  17. 4: dict(name='kpt-4', id=4, color=[255, 0, 0], type='', swap='kpt-28'),
  18. 5: dict(name='kpt-5', id=5, color=[255, 0, 0], type='', swap='kpt-27'),
  19. 6: dict(name='kpt-6', id=6, color=[255, 0, 0], type='', swap='kpt-26'),
  20. 7: dict(name='kpt-7', id=7, color=[255, 0, 0], type='', swap='kpt-25'),
  21. 8: dict(name='kpt-8', id=8, color=[255, 0, 0], type='', swap='kpt-24'),
  22. 9: dict(name='kpt-9', id=9, color=[255, 0, 0], type='', swap='kpt-23'),
  23. 10:
  24. dict(name='kpt-10', id=10, color=[255, 0, 0], type='', swap='kpt-22'),
  25. 11:
  26. dict(name='kpt-11', id=11, color=[255, 0, 0], type='', swap='kpt-21'),
  27. 12:
  28. dict(name='kpt-12', id=12, color=[255, 0, 0], type='', swap='kpt-20'),
  29. 13:
  30. dict(name='kpt-13', id=13, color=[255, 0, 0], type='', swap='kpt-19'),
  31. 14:
  32. dict(name='kpt-14', id=14, color=[255, 0, 0], type='', swap='kpt-18'),
  33. 15:
  34. dict(name='kpt-15', id=15, color=[255, 0, 0], type='', swap='kpt-17'),
  35. 16: dict(name='kpt-16', id=16, color=[255, 0, 0], type='', swap=''),
  36. 17:
  37. dict(name='kpt-17', id=17, color=[255, 0, 0], type='', swap='kpt-15'),
  38. 18:
  39. dict(name='kpt-18', id=18, color=[255, 0, 0], type='', swap='kpt-14'),
  40. 19:
  41. dict(name='kpt-19', id=19, color=[255, 0, 0], type='', swap='kpt-13'),
  42. 20:
  43. dict(name='kpt-20', id=20, color=[255, 0, 0], type='', swap='kpt-12'),
  44. 21:
  45. dict(name='kpt-21', id=21, color=[255, 0, 0], type='', swap='kpt-11'),
  46. 22:
  47. dict(name='kpt-22', id=22, color=[255, 0, 0], type='', swap='kpt-10'),
  48. 23:
  49. dict(name='kpt-23', id=23, color=[255, 0, 0], type='', swap='kpt-9'),
  50. 24:
  51. dict(name='kpt-24', id=24, color=[255, 0, 0], type='', swap='kpt-8'),
  52. 25:
  53. dict(name='kpt-25', id=25, color=[255, 0, 0], type='', swap='kpt-7'),
  54. 26:
  55. dict(name='kpt-26', id=26, color=[255, 0, 0], type='', swap='kpt-6'),
  56. 27:
  57. dict(name='kpt-27', id=27, color=[255, 0, 0], type='', swap='kpt-5'),
  58. 28:
  59. dict(name='kpt-28', id=28, color=[255, 0, 0], type='', swap='kpt-4'),
  60. 29:
  61. dict(name='kpt-29', id=29, color=[255, 0, 0], type='', swap='kpt-3'),
  62. 30:
  63. dict(name='kpt-30', id=30, color=[255, 0, 0], type='', swap='kpt-2'),
  64. 31:
  65. dict(name='kpt-31', id=31, color=[255, 0, 0], type='', swap='kpt-1'),
  66. 32:
  67. dict(name='kpt-32', id=32, color=[255, 0, 0], type='', swap='kpt-0'),
  68. 33:
  69. dict(name='kpt-33', id=33, color=[255, 0, 0], type='', swap='kpt-46'),
  70. 34:
  71. dict(name='kpt-34', id=34, color=[255, 0, 0], type='', swap='kpt-45'),
  72. 35:
  73. dict(name='kpt-35', id=35, color=[255, 0, 0], type='', swap='kpt-44'),
  74. 36:
  75. dict(name='kpt-36', id=36, color=[255, 0, 0], type='', swap='kpt-43'),
  76. 37: dict(
  77. name='kpt-37', id=37, color=[255, 0, 0], type='', swap='kpt-42'),
  78. 38: dict(
  79. name='kpt-38', id=38, color=[255, 0, 0], type='', swap='kpt-50'),
  80. 39: dict(
  81. name='kpt-39', id=39, color=[255, 0, 0], type='', swap='kpt-49'),
  82. 40: dict(
  83. name='kpt-40', id=40, color=[255, 0, 0], type='', swap='kpt-48'),
  84. 41: dict(
  85. name='kpt-41', id=41, color=[255, 0, 0], type='', swap='kpt-47'),
  86. 42: dict(
  87. name='kpt-42', id=42, color=[255, 0, 0], type='', swap='kpt-37'),
  88. 43: dict(
  89. name='kpt-43', id=43, color=[255, 0, 0], type='', swap='kpt-36'),
  90. 44: dict(
  91. name='kpt-44', id=44, color=[255, 0, 0], type='', swap='kpt-35'),
  92. 45: dict(
  93. name='kpt-45', id=45, color=[255, 0, 0], type='', swap='kpt-34'),
  94. 46: dict(
  95. name='kpt-46', id=46, color=[255, 0, 0], type='', swap='kpt-33'),
  96. 47: dict(
  97. name='kpt-47', id=47, color=[255, 0, 0], type='', swap='kpt-41'),
  98. 48: dict(
  99. name='kpt-48', id=48, color=[255, 0, 0], type='', swap='kpt-40'),
  100. 49: dict(
  101. name='kpt-49', id=49, color=[255, 0, 0], type='', swap='kpt-39'),
  102. 50: dict(
  103. name='kpt-50', id=50, color=[255, 0, 0], type='', swap='kpt-38'),
  104. 51: dict(name='kpt-51', id=51, color=[255, 0, 0], type='', swap=''),
  105. 52: dict(name='kpt-52', id=52, color=[255, 0, 0], type='', swap=''),
  106. 53: dict(name='kpt-53', id=53, color=[255, 0, 0], type='', swap=''),
  107. 54: dict(name='kpt-54', id=54, color=[255, 0, 0], type='', swap=''),
  108. 55: dict(
  109. name='kpt-55', id=55, color=[255, 0, 0], type='', swap='kpt-59'),
  110. 56: dict(
  111. name='kpt-56', id=56, color=[255, 0, 0], type='', swap='kpt-58'),
  112. 57: dict(name='kpt-57', id=57, color=[255, 0, 0], type='', swap=''),
  113. 58: dict(
  114. name='kpt-58', id=58, color=[255, 0, 0], type='', swap='kpt-56'),
  115. 59: dict(
  116. name='kpt-59', id=59, color=[255, 0, 0], type='', swap='kpt-55'),
  117. 60: dict(
  118. name='kpt-60', id=60, color=[255, 0, 0], type='', swap='kpt-72'),
  119. 61: dict(
  120. name='kpt-61', id=61, color=[255, 0, 0], type='', swap='kpt-71'),
  121. 62: dict(
  122. name='kpt-62', id=62, color=[255, 0, 0], type='', swap='kpt-70'),
  123. 63: dict(
  124. name='kpt-63', id=63, color=[255, 0, 0], type='', swap='kpt-69'),
  125. 64: dict(
  126. name='kpt-64', id=64, color=[255, 0, 0], type='', swap='kpt-68'),
  127. 65: dict(
  128. name='kpt-65', id=65, color=[255, 0, 0], type='', swap='kpt-75'),
  129. 66: dict(
  130. name='kpt-66', id=66, color=[255, 0, 0], type='', swap='kpt-74'),
  131. 67: dict(
  132. name='kpt-67', id=67, color=[255, 0, 0], type='', swap='kpt-73'),
  133. 68: dict(
  134. name='kpt-68', id=68, color=[255, 0, 0], type='', swap='kpt-64'),
  135. 69: dict(
  136. name='kpt-69', id=69, color=[255, 0, 0], type='', swap='kpt-63'),
  137. 70: dict(
  138. name='kpt-70', id=70, color=[255, 0, 0], type='', swap='kpt-62'),
  139. 71: dict(
  140. name='kpt-71', id=71, color=[255, 0, 0], type='', swap='kpt-61'),
  141. 72: dict(
  142. name='kpt-72', id=72, color=[255, 0, 0], type='', swap='kpt-60'),
  143. 73: dict(
  144. name='kpt-73', id=73, color=[255, 0, 0], type='', swap='kpt-67'),
  145. 74: dict(
  146. name='kpt-74', id=74, color=[255, 0, 0], type='', swap='kpt-66'),
  147. 75: dict(
  148. name='kpt-75', id=75, color=[255, 0, 0], type='', swap='kpt-65'),
  149. 76: dict(
  150. name='kpt-76', id=76, color=[255, 0, 0], type='', swap='kpt-82'),
  151. 77: dict(
  152. name='kpt-77', id=77, color=[255, 0, 0], type='', swap='kpt-81'),
  153. 78: dict(
  154. name='kpt-78', id=78, color=[255, 0, 0], type='', swap='kpt-80'),
  155. 79: dict(name='kpt-79', id=79, color=[255, 0, 0], type='', swap=''),
  156. 80: dict(
  157. name='kpt-80', id=80, color=[255, 0, 0], type='', swap='kpt-78'),
  158. 81: dict(
  159. name='kpt-81', id=81, color=[255, 0, 0], type='', swap='kpt-77'),
  160. 82: dict(
  161. name='kpt-82', id=82, color=[255, 0, 0], type='', swap='kpt-76'),
  162. 83: dict(
  163. name='kpt-83', id=83, color=[255, 0, 0], type='', swap='kpt-87'),
  164. 84: dict(
  165. name='kpt-84', id=84, color=[255, 0, 0], type='', swap='kpt-86'),
  166. 85: dict(name='kpt-85', id=85, color=[255, 0, 0], type='', swap=''),
  167. 86: dict(
  168. name='kpt-86', id=86, color=[255, 0, 0], type='', swap='kpt-84'),
  169. 87: dict(
  170. name='kpt-87', id=87, color=[255, 0, 0], type='', swap='kpt-83'),
  171. 88: dict(
  172. name='kpt-88', id=88, color=[255, 0, 0], type='', swap='kpt-92'),
  173. 89: dict(
  174. name='kpt-89', id=89, color=[255, 0, 0], type='', swap='kpt-91'),
  175. 90: dict(name='kpt-90', id=90, color=[255, 0, 0], type='', swap=''),
  176. 91: dict(
  177. name='kpt-91', id=91, color=[255, 0, 0], type='', swap='kpt-89'),
  178. 92: dict(
  179. name='kpt-92', id=92, color=[255, 0, 0], type='', swap='kpt-88'),
  180. 93: dict(
  181. name='kpt-93', id=93, color=[255, 0, 0], type='', swap='kpt-95'),
  182. 94: dict(name='kpt-94', id=94, color=[255, 0, 0], type='', swap=''),
  183. 95: dict(
  184. name='kpt-95', id=95, color=[255, 0, 0], type='', swap='kpt-93'),
  185. 96: dict(
  186. name='kpt-96', id=96, color=[255, 0, 0], type='', swap='kpt-97'),
  187. 97: dict(
  188. name='kpt-97', id=97, color=[255, 0, 0], type='', swap='kpt-96')
  189. },
  190. skeleton_info={},
  191. joint_weights=[1.] * 98,
  192. sigmas=[])