coco_wholebody_hand.py 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147
  1. dataset_info = dict(
  2. dataset_name='coco_wholebody_hand',
  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='wrist', id=0, color=[255, 255, 255], type='', swap=''),
  16. 1:
  17. dict(name='thumb1', id=1, color=[255, 128, 0], type='', swap=''),
  18. 2:
  19. dict(name='thumb2', id=2, color=[255, 128, 0], type='', swap=''),
  20. 3:
  21. dict(name='thumb3', id=3, color=[255, 128, 0], type='', swap=''),
  22. 4:
  23. dict(name='thumb4', id=4, color=[255, 128, 0], type='', swap=''),
  24. 5:
  25. dict(
  26. name='forefinger1', id=5, color=[255, 153, 255], type='', swap=''),
  27. 6:
  28. dict(
  29. name='forefinger2', id=6, color=[255, 153, 255], type='', swap=''),
  30. 7:
  31. dict(
  32. name='forefinger3', id=7, color=[255, 153, 255], type='', swap=''),
  33. 8:
  34. dict(
  35. name='forefinger4', id=8, color=[255, 153, 255], type='', swap=''),
  36. 9:
  37. dict(
  38. name='middle_finger1',
  39. id=9,
  40. color=[102, 178, 255],
  41. type='',
  42. swap=''),
  43. 10:
  44. dict(
  45. name='middle_finger2',
  46. id=10,
  47. color=[102, 178, 255],
  48. type='',
  49. swap=''),
  50. 11:
  51. dict(
  52. name='middle_finger3',
  53. id=11,
  54. color=[102, 178, 255],
  55. type='',
  56. swap=''),
  57. 12:
  58. dict(
  59. name='middle_finger4',
  60. id=12,
  61. color=[102, 178, 255],
  62. type='',
  63. swap=''),
  64. 13:
  65. dict(
  66. name='ring_finger1', id=13, color=[255, 51, 51], type='', swap=''),
  67. 14:
  68. dict(
  69. name='ring_finger2', id=14, color=[255, 51, 51], type='', swap=''),
  70. 15:
  71. dict(
  72. name='ring_finger3', id=15, color=[255, 51, 51], type='', swap=''),
  73. 16:
  74. dict(
  75. name='ring_finger4', id=16, color=[255, 51, 51], type='', swap=''),
  76. 17:
  77. dict(name='pinky_finger1', id=17, color=[0, 255, 0], type='', swap=''),
  78. 18:
  79. dict(name='pinky_finger2', id=18, color=[0, 255, 0], type='', swap=''),
  80. 19:
  81. dict(name='pinky_finger3', id=19, color=[0, 255, 0], type='', swap=''),
  82. 20:
  83. dict(name='pinky_finger4', id=20, color=[0, 255, 0], type='', swap='')
  84. },
  85. skeleton_info={
  86. 0:
  87. dict(link=('wrist', 'thumb1'), id=0, color=[255, 128, 0]),
  88. 1:
  89. dict(link=('thumb1', 'thumb2'), id=1, color=[255, 128, 0]),
  90. 2:
  91. dict(link=('thumb2', 'thumb3'), id=2, color=[255, 128, 0]),
  92. 3:
  93. dict(link=('thumb3', 'thumb4'), id=3, color=[255, 128, 0]),
  94. 4:
  95. dict(link=('wrist', 'forefinger1'), id=4, color=[255, 153, 255]),
  96. 5:
  97. dict(link=('forefinger1', 'forefinger2'), id=5, color=[255, 153, 255]),
  98. 6:
  99. dict(link=('forefinger2', 'forefinger3'), id=6, color=[255, 153, 255]),
  100. 7:
  101. dict(link=('forefinger3', 'forefinger4'), id=7, color=[255, 153, 255]),
  102. 8:
  103. dict(link=('wrist', 'middle_finger1'), id=8, color=[102, 178, 255]),
  104. 9:
  105. dict(
  106. link=('middle_finger1', 'middle_finger2'),
  107. id=9,
  108. color=[102, 178, 255]),
  109. 10:
  110. dict(
  111. link=('middle_finger2', 'middle_finger3'),
  112. id=10,
  113. color=[102, 178, 255]),
  114. 11:
  115. dict(
  116. link=('middle_finger3', 'middle_finger4'),
  117. id=11,
  118. color=[102, 178, 255]),
  119. 12:
  120. dict(link=('wrist', 'ring_finger1'), id=12, color=[255, 51, 51]),
  121. 13:
  122. dict(
  123. link=('ring_finger1', 'ring_finger2'), id=13, color=[255, 51, 51]),
  124. 14:
  125. dict(
  126. link=('ring_finger2', 'ring_finger3'), id=14, color=[255, 51, 51]),
  127. 15:
  128. dict(
  129. link=('ring_finger3', 'ring_finger4'), id=15, color=[255, 51, 51]),
  130. 16:
  131. dict(link=('wrist', 'pinky_finger1'), id=16, color=[0, 255, 0]),
  132. 17:
  133. dict(
  134. link=('pinky_finger1', 'pinky_finger2'), id=17, color=[0, 255, 0]),
  135. 18:
  136. dict(
  137. link=('pinky_finger2', 'pinky_finger3'), id=18, color=[0, 255, 0]),
  138. 19:
  139. dict(
  140. link=('pinky_finger3', 'pinky_finger4'), id=19, color=[0, 255, 0])
  141. },
  142. joint_weights=[1.] * 21,
  143. sigmas=[
  144. 0.029, 0.022, 0.035, 0.037, 0.047, 0.026, 0.025, 0.024, 0.035, 0.018,
  145. 0.024, 0.022, 0.026, 0.017, 0.021, 0.021, 0.032, 0.02, 0.019, 0.022,
  146. 0.031
  147. ])