horse10.py 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201
  1. dataset_info = dict(
  2. dataset_name='horse10',
  3. paper_info=dict(
  4. author='Mathis, Alexander and Biasi, Thomas and '
  5. 'Schneider, Steffen and '
  6. 'Yuksekgonul, Mert and Rogers, Byron and '
  7. 'Bethge, Matthias and '
  8. 'Mathis, Mackenzie W',
  9. title='Pretraining boosts out-of-domain robustness '
  10. 'for pose estimation',
  11. container='Proceedings of the IEEE/CVF Winter Conference on '
  12. 'Applications of Computer Vision',
  13. year='2021',
  14. homepage='http://www.mackenziemathislab.org/horse10',
  15. ),
  16. keypoint_info={
  17. 0:
  18. dict(name='Nose', id=0, color=[255, 153, 255], type='upper', swap=''),
  19. 1:
  20. dict(name='Eye', id=1, color=[255, 153, 255], type='upper', swap=''),
  21. 2:
  22. dict(
  23. name='Nearknee',
  24. id=2,
  25. color=[255, 102, 255],
  26. type='upper',
  27. swap=''),
  28. 3:
  29. dict(
  30. name='Nearfrontfetlock',
  31. id=3,
  32. color=[255, 102, 255],
  33. type='upper',
  34. swap=''),
  35. 4:
  36. dict(
  37. name='Nearfrontfoot',
  38. id=4,
  39. color=[255, 102, 255],
  40. type='upper',
  41. swap=''),
  42. 5:
  43. dict(
  44. name='Offknee', id=5, color=[255, 102, 255], type='upper',
  45. swap=''),
  46. 6:
  47. dict(
  48. name='Offfrontfetlock',
  49. id=6,
  50. color=[255, 102, 255],
  51. type='upper',
  52. swap=''),
  53. 7:
  54. dict(
  55. name='Offfrontfoot',
  56. id=7,
  57. color=[255, 102, 255],
  58. type='upper',
  59. swap=''),
  60. 8:
  61. dict(
  62. name='Shoulder',
  63. id=8,
  64. color=[255, 153, 255],
  65. type='upper',
  66. swap=''),
  67. 9:
  68. dict(
  69. name='Midshoulder',
  70. id=9,
  71. color=[255, 153, 255],
  72. type='upper',
  73. swap=''),
  74. 10:
  75. dict(
  76. name='Elbow', id=10, color=[255, 153, 255], type='upper', swap=''),
  77. 11:
  78. dict(
  79. name='Girth', id=11, color=[255, 153, 255], type='upper', swap=''),
  80. 12:
  81. dict(
  82. name='Wither', id=12, color=[255, 153, 255], type='upper',
  83. swap=''),
  84. 13:
  85. dict(
  86. name='Nearhindhock',
  87. id=13,
  88. color=[255, 51, 255],
  89. type='lower',
  90. swap=''),
  91. 14:
  92. dict(
  93. name='Nearhindfetlock',
  94. id=14,
  95. color=[255, 51, 255],
  96. type='lower',
  97. swap=''),
  98. 15:
  99. dict(
  100. name='Nearhindfoot',
  101. id=15,
  102. color=[255, 51, 255],
  103. type='lower',
  104. swap=''),
  105. 16:
  106. dict(name='Hip', id=16, color=[255, 153, 255], type='lower', swap=''),
  107. 17:
  108. dict(
  109. name='Stifle', id=17, color=[255, 153, 255], type='lower',
  110. swap=''),
  111. 18:
  112. dict(
  113. name='Offhindhock',
  114. id=18,
  115. color=[255, 51, 255],
  116. type='lower',
  117. swap=''),
  118. 19:
  119. dict(
  120. name='Offhindfetlock',
  121. id=19,
  122. color=[255, 51, 255],
  123. type='lower',
  124. swap=''),
  125. 20:
  126. dict(
  127. name='Offhindfoot',
  128. id=20,
  129. color=[255, 51, 255],
  130. type='lower',
  131. swap=''),
  132. 21:
  133. dict(
  134. name='Ischium',
  135. id=21,
  136. color=[255, 153, 255],
  137. type='lower',
  138. swap='')
  139. },
  140. skeleton_info={
  141. 0:
  142. dict(link=('Nose', 'Eye'), id=0, color=[255, 153, 255]),
  143. 1:
  144. dict(link=('Eye', 'Wither'), id=1, color=[255, 153, 255]),
  145. 2:
  146. dict(link=('Wither', 'Hip'), id=2, color=[255, 153, 255]),
  147. 3:
  148. dict(link=('Hip', 'Ischium'), id=3, color=[255, 153, 255]),
  149. 4:
  150. dict(link=('Ischium', 'Stifle'), id=4, color=[255, 153, 255]),
  151. 5:
  152. dict(link=('Stifle', 'Girth'), id=5, color=[255, 153, 255]),
  153. 6:
  154. dict(link=('Girth', 'Elbow'), id=6, color=[255, 153, 255]),
  155. 7:
  156. dict(link=('Elbow', 'Shoulder'), id=7, color=[255, 153, 255]),
  157. 8:
  158. dict(link=('Shoulder', 'Midshoulder'), id=8, color=[255, 153, 255]),
  159. 9:
  160. dict(link=('Midshoulder', 'Wither'), id=9, color=[255, 153, 255]),
  161. 10:
  162. dict(
  163. link=('Nearknee', 'Nearfrontfetlock'),
  164. id=10,
  165. color=[255, 102, 255]),
  166. 11:
  167. dict(
  168. link=('Nearfrontfetlock', 'Nearfrontfoot'),
  169. id=11,
  170. color=[255, 102, 255]),
  171. 12:
  172. dict(
  173. link=('Offknee', 'Offfrontfetlock'), id=12, color=[255, 102, 255]),
  174. 13:
  175. dict(
  176. link=('Offfrontfetlock', 'Offfrontfoot'),
  177. id=13,
  178. color=[255, 102, 255]),
  179. 14:
  180. dict(
  181. link=('Nearhindhock', 'Nearhindfetlock'),
  182. id=14,
  183. color=[255, 51, 255]),
  184. 15:
  185. dict(
  186. link=('Nearhindfetlock', 'Nearhindfoot'),
  187. id=15,
  188. color=[255, 51, 255]),
  189. 16:
  190. dict(
  191. link=('Offhindhock', 'Offhindfetlock'),
  192. id=16,
  193. color=[255, 51, 255]),
  194. 17:
  195. dict(
  196. link=('Offhindfetlock', 'Offhindfoot'),
  197. id=17,
  198. color=[255, 51, 255])
  199. },
  200. joint_weights=[1.] * 22,
  201. sigmas=[])