panoptic_body3d.py 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160
  1. dataset_info = dict(
  2. dataset_name='panoptic_pose_3d',
  3. paper_info=dict(
  4. author='Joo, Hanbyul and Simon, Tomas and Li, Xulong'
  5. 'and Liu, Hao and Tan, Lei and Gui, Lin and Banerjee, Sean'
  6. 'and Godisart, Timothy and Nabbe, Bart and Matthews, Iain'
  7. 'and Kanade, Takeo and Nobuhara, Shohei and Sheikh, Yaser',
  8. title='Panoptic Studio: A Massively Multiview System '
  9. 'for Interaction Motion Capture',
  10. container='IEEE Transactions on Pattern Analysis'
  11. ' and Machine Intelligence',
  12. year='2017',
  13. homepage='http://domedb.perception.cs.cmu.edu',
  14. ),
  15. keypoint_info={
  16. 0:
  17. dict(name='neck', id=0, color=[51, 153, 255], type='upper', swap=''),
  18. 1:
  19. dict(name='nose', id=1, color=[51, 153, 255], type='upper', swap=''),
  20. 2:
  21. dict(name='mid_hip', id=2, color=[0, 255, 0], type='lower', swap=''),
  22. 3:
  23. dict(
  24. name='left_shoulder',
  25. id=3,
  26. color=[0, 255, 0],
  27. type='upper',
  28. swap='right_shoulder'),
  29. 4:
  30. dict(
  31. name='left_elbow',
  32. id=4,
  33. color=[0, 255, 0],
  34. type='upper',
  35. swap='right_elbow'),
  36. 5:
  37. dict(
  38. name='left_wrist',
  39. id=5,
  40. color=[0, 255, 0],
  41. type='upper',
  42. swap='right_wrist'),
  43. 6:
  44. dict(
  45. name='left_hip',
  46. id=6,
  47. color=[0, 255, 0],
  48. type='lower',
  49. swap='right_hip'),
  50. 7:
  51. dict(
  52. name='left_knee',
  53. id=7,
  54. color=[0, 255, 0],
  55. type='lower',
  56. swap='right_knee'),
  57. 8:
  58. dict(
  59. name='left_ankle',
  60. id=8,
  61. color=[0, 255, 0],
  62. type='lower',
  63. swap='right_ankle'),
  64. 9:
  65. dict(
  66. name='right_shoulder',
  67. id=9,
  68. color=[255, 128, 0],
  69. type='upper',
  70. swap='left_shoulder'),
  71. 10:
  72. dict(
  73. name='right_elbow',
  74. id=10,
  75. color=[255, 128, 0],
  76. type='upper',
  77. swap='left_elbow'),
  78. 11:
  79. dict(
  80. name='right_wrist',
  81. id=11,
  82. color=[255, 128, 0],
  83. type='upper',
  84. swap='left_wrist'),
  85. 12:
  86. dict(
  87. name='right_hip',
  88. id=12,
  89. color=[255, 128, 0],
  90. type='lower',
  91. swap='left_hip'),
  92. 13:
  93. dict(
  94. name='right_knee',
  95. id=13,
  96. color=[255, 128, 0],
  97. type='lower',
  98. swap='left_knee'),
  99. 14:
  100. dict(
  101. name='right_ankle',
  102. id=14,
  103. color=[255, 128, 0],
  104. type='lower',
  105. swap='left_ankle'),
  106. 15:
  107. dict(
  108. name='left_eye',
  109. id=15,
  110. color=[51, 153, 255],
  111. type='upper',
  112. swap='right_eye'),
  113. 16:
  114. dict(
  115. name='left_ear',
  116. id=16,
  117. color=[51, 153, 255],
  118. type='upper',
  119. swap='right_ear'),
  120. 17:
  121. dict(
  122. name='right_eye',
  123. id=17,
  124. color=[51, 153, 255],
  125. type='upper',
  126. swap='left_eye'),
  127. 18:
  128. dict(
  129. name='right_ear',
  130. id=18,
  131. color=[51, 153, 255],
  132. type='upper',
  133. swap='left_ear')
  134. },
  135. skeleton_info={
  136. 0: dict(link=('nose', 'neck'), id=0, color=[51, 153, 255]),
  137. 1: dict(link=('neck', 'left_shoulder'), id=1, color=[0, 255, 0]),
  138. 2: dict(link=('neck', 'right_shoulder'), id=2, color=[255, 128, 0]),
  139. 3: dict(link=('left_shoulder', 'left_elbow'), id=3, color=[0, 255, 0]),
  140. 4: dict(
  141. link=('right_shoulder', 'right_elbow'), id=4, color=[255, 128, 0]),
  142. 5: dict(link=('left_elbow', 'left_wrist'), id=5, color=[0, 255, 0]),
  143. 6:
  144. dict(link=('right_elbow', 'right_wrist'), id=6, color=[255, 128, 0]),
  145. 7: dict(link=('left_ankle', 'left_knee'), id=7, color=[0, 255, 0]),
  146. 8: dict(link=('left_knee', 'left_hip'), id=8, color=[0, 255, 0]),
  147. 9: dict(link=('right_ankle', 'right_knee'), id=9, color=[255, 128, 0]),
  148. 10: dict(link=('right_knee', 'right_hip'), id=10, color=[255, 128, 0]),
  149. 11: dict(link=('mid_hip', 'left_hip'), id=11, color=[0, 255, 0]),
  150. 12: dict(link=('mid_hip', 'right_hip'), id=12, color=[255, 128, 0]),
  151. 13: dict(link=('mid_hip', 'neck'), id=13, color=[51, 153, 255]),
  152. },
  153. joint_weights=[
  154. 1.0, 1.0, 1.0, 1.0, 1.2, 1.5, 1.0, 1.2, 1.5, 1.0, 1.2, 1.5, 1.0, 1.2,
  155. 1.5, 1.0, 1.0, 1.0, 1.0
  156. ],
  157. sigmas=[
  158. 0.026, 0.026, 0.107, 0.079, 0.072, 0.062, 0.107, 0.087, 0.089, 0.079,
  159. 0.072, 0.062, 0.107, 0.087, 0.089, 0.025, 0.035, 0.025, 0.035
  160. ])