coco_wholebody_face.py 7.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154
  1. dataset_info = dict(
  2. dataset_name='coco_wholebody_face',
  3. paper_info=dict(
  4. author='Jin, Sheng and Xu, Lumin and Xu, Jin and '
  5. 'Wang, Can and Liu, Wentao and '
  6. 'Qian, Chen and Ouyang, Wanli and Luo, Ping',
  7. title='Whole-Body Human Pose Estimation in the Wild',
  8. container='Proceedings of the European '
  9. 'Conference on Computer Vision (ECCV)',
  10. year='2020',
  11. homepage='https://github.com/jin-s13/COCO-WholeBody/',
  12. ),
  13. keypoint_info={
  14. 0:
  15. dict(name='face-0', id=0, color=[255, 0, 0], type='', swap='face-16'),
  16. 1:
  17. dict(name='face-1', id=1, color=[255, 0, 0], type='', swap='face-15'),
  18. 2:
  19. dict(name='face-2', id=2, color=[255, 0, 0], type='', swap='face-14'),
  20. 3:
  21. dict(name='face-3', id=3, color=[255, 0, 0], type='', swap='face-13'),
  22. 4:
  23. dict(name='face-4', id=4, color=[255, 0, 0], type='', swap='face-12'),
  24. 5:
  25. dict(name='face-5', id=5, color=[255, 0, 0], type='', swap='face-11'),
  26. 6:
  27. dict(name='face-6', id=6, color=[255, 0, 0], type='', swap='face-10'),
  28. 7:
  29. dict(name='face-7', id=7, color=[255, 0, 0], type='', swap='face-9'),
  30. 8: dict(name='face-8', id=8, color=[255, 0, 0], type='', swap=''),
  31. 9:
  32. dict(name='face-9', id=9, color=[255, 0, 0], type='', swap='face-7'),
  33. 10:
  34. dict(name='face-10', id=10, color=[255, 0, 0], type='', swap='face-6'),
  35. 11:
  36. dict(name='face-11', id=11, color=[255, 0, 0], type='', swap='face-5'),
  37. 12:
  38. dict(name='face-12', id=12, color=[255, 0, 0], type='', swap='face-4'),
  39. 13:
  40. dict(name='face-13', id=13, color=[255, 0, 0], type='', swap='face-3'),
  41. 14:
  42. dict(name='face-14', id=14, color=[255, 0, 0], type='', swap='face-2'),
  43. 15:
  44. dict(name='face-15', id=15, color=[255, 0, 0], type='', swap='face-1'),
  45. 16:
  46. dict(name='face-16', id=16, color=[255, 0, 0], type='', swap='face-0'),
  47. 17: dict(
  48. name='face-17', id=17, color=[255, 0, 0], type='', swap='face-26'),
  49. 18: dict(
  50. name='face-18', id=18, color=[255, 0, 0], type='', swap='face-25'),
  51. 19: dict(
  52. name='face-19', id=19, color=[255, 0, 0], type='', swap='face-24'),
  53. 20: dict(
  54. name='face-20', id=20, color=[255, 0, 0], type='', swap='face-23'),
  55. 21: dict(
  56. name='face-21', id=21, color=[255, 0, 0], type='', swap='face-22'),
  57. 22: dict(
  58. name='face-22', id=22, color=[255, 0, 0], type='', swap='face-21'),
  59. 23: dict(
  60. name='face-23', id=23, color=[255, 0, 0], type='', swap='face-20'),
  61. 24: dict(
  62. name='face-24', id=24, color=[255, 0, 0], type='', swap='face-19'),
  63. 25: dict(
  64. name='face-25', id=25, color=[255, 0, 0], type='', swap='face-18'),
  65. 26: dict(
  66. name='face-26', id=26, color=[255, 0, 0], type='', swap='face-17'),
  67. 27: dict(name='face-27', id=27, color=[255, 0, 0], type='', swap=''),
  68. 28: dict(name='face-28', id=28, color=[255, 0, 0], type='', swap=''),
  69. 29: dict(name='face-29', id=29, color=[255, 0, 0], type='', swap=''),
  70. 30: dict(name='face-30', id=30, color=[255, 0, 0], type='', swap=''),
  71. 31: dict(
  72. name='face-31', id=31, color=[255, 0, 0], type='', swap='face-35'),
  73. 32: dict(
  74. name='face-32', id=32, color=[255, 0, 0], type='', swap='face-34'),
  75. 33: dict(name='face-33', id=33, color=[255, 0, 0], type='', swap=''),
  76. 34: dict(
  77. name='face-34', id=34, color=[255, 0, 0], type='', swap='face-32'),
  78. 35: dict(
  79. name='face-35', id=35, color=[255, 0, 0], type='', swap='face-31'),
  80. 36: dict(
  81. name='face-36', id=36, color=[255, 0, 0], type='', swap='face-45'),
  82. 37: dict(
  83. name='face-37', id=37, color=[255, 0, 0], type='', swap='face-44'),
  84. 38: dict(
  85. name='face-38', id=38, color=[255, 0, 0], type='', swap='face-43'),
  86. 39: dict(
  87. name='face-39', id=39, color=[255, 0, 0], type='', swap='face-42'),
  88. 40: dict(
  89. name='face-40', id=40, color=[255, 0, 0], type='', swap='face-47'),
  90. 41: dict(
  91. name='face-41', id=41, color=[255, 0, 0], type='', swap='face-46'),
  92. 42: dict(
  93. name='face-42', id=42, color=[255, 0, 0], type='', swap='face-39'),
  94. 43: dict(
  95. name='face-43', id=43, color=[255, 0, 0], type='', swap='face-38'),
  96. 44: dict(
  97. name='face-44', id=44, color=[255, 0, 0], type='', swap='face-37'),
  98. 45: dict(
  99. name='face-45', id=45, color=[255, 0, 0], type='', swap='face-36'),
  100. 46: dict(
  101. name='face-46', id=46, color=[255, 0, 0], type='', swap='face-41'),
  102. 47: dict(
  103. name='face-47', id=47, color=[255, 0, 0], type='', swap='face-40'),
  104. 48: dict(
  105. name='face-48', id=48, color=[255, 0, 0], type='', swap='face-54'),
  106. 49: dict(
  107. name='face-49', id=49, color=[255, 0, 0], type='', swap='face-53'),
  108. 50: dict(
  109. name='face-50', id=50, color=[255, 0, 0], type='', swap='face-52'),
  110. 51: dict(name='face-51', id=52, color=[255, 0, 0], type='', swap=''),
  111. 52: dict(
  112. name='face-52', id=52, color=[255, 0, 0], type='', swap='face-50'),
  113. 53: dict(
  114. name='face-53', id=53, color=[255, 0, 0], type='', swap='face-49'),
  115. 54: dict(
  116. name='face-54', id=54, color=[255, 0, 0], type='', swap='face-48'),
  117. 55: dict(
  118. name='face-55', id=55, color=[255, 0, 0], type='', swap='face-59'),
  119. 56: dict(
  120. name='face-56', id=56, color=[255, 0, 0], type='', swap='face-58'),
  121. 57: dict(name='face-57', id=57, color=[255, 0, 0], type='', swap=''),
  122. 58: dict(
  123. name='face-58', id=58, color=[255, 0, 0], type='', swap='face-56'),
  124. 59: dict(
  125. name='face-59', id=59, color=[255, 0, 0], type='', swap='face-55'),
  126. 60: dict(
  127. name='face-60', id=60, color=[255, 0, 0], type='', swap='face-64'),
  128. 61: dict(
  129. name='face-61', id=61, color=[255, 0, 0], type='', swap='face-63'),
  130. 62: dict(name='face-62', id=62, color=[255, 0, 0], type='', swap=''),
  131. 63: dict(
  132. name='face-63', id=63, color=[255, 0, 0], type='', swap='face-61'),
  133. 64: dict(
  134. name='face-64', id=64, color=[255, 0, 0], type='', swap='face-60'),
  135. 65: dict(
  136. name='face-65', id=65, color=[255, 0, 0], type='', swap='face-67'),
  137. 66: dict(name='face-66', id=66, color=[255, 0, 0], type='', swap=''),
  138. 67: dict(
  139. name='face-67', id=67, color=[255, 0, 0], type='', swap='face-65')
  140. },
  141. skeleton_info={},
  142. joint_weights=[1.] * 68,
  143. # 'https://github.com/jin-s13/COCO-WholeBody/blob/master/'
  144. # 'evaluation/myeval_wholebody.py#L177'
  145. sigmas=[
  146. 0.042, 0.043, 0.044, 0.043, 0.040, 0.035, 0.031, 0.025, 0.020, 0.023,
  147. 0.029, 0.032, 0.037, 0.038, 0.043, 0.041, 0.045, 0.013, 0.012, 0.011,
  148. 0.011, 0.012, 0.012, 0.011, 0.011, 0.013, 0.015, 0.009, 0.007, 0.007,
  149. 0.007, 0.012, 0.009, 0.008, 0.016, 0.010, 0.017, 0.011, 0.009, 0.011,
  150. 0.009, 0.007, 0.013, 0.008, 0.011, 0.012, 0.010, 0.034, 0.008, 0.008,
  151. 0.009, 0.008, 0.008, 0.007, 0.010, 0.008, 0.009, 0.009, 0.009, 0.007,
  152. 0.007, 0.008, 0.011, 0.008, 0.008, 0.008, 0.01, 0.008
  153. ])