interhand3d.py 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487
  1. dataset_info = dict(
  2. dataset_name='interhand3d',
  3. paper_info=dict(
  4. author='Moon, Gyeongsik and Yu, Shoou-I and Wen, He and '
  5. 'Shiratori, Takaaki and Lee, Kyoung Mu',
  6. title='InterHand2.6M: A dataset and baseline for 3D '
  7. 'interacting hand pose estimation from a single RGB image',
  8. container='arXiv',
  9. year='2020',
  10. homepage='https://mks0601.github.io/InterHand2.6M/',
  11. ),
  12. keypoint_info={
  13. 0:
  14. dict(
  15. name='right_thumb4',
  16. id=0,
  17. color=[255, 128, 0],
  18. type='',
  19. swap='left_thumb4'),
  20. 1:
  21. dict(
  22. name='right_thumb3',
  23. id=1,
  24. color=[255, 128, 0],
  25. type='',
  26. swap='left_thumb3'),
  27. 2:
  28. dict(
  29. name='right_thumb2',
  30. id=2,
  31. color=[255, 128, 0],
  32. type='',
  33. swap='left_thumb2'),
  34. 3:
  35. dict(
  36. name='right_thumb1',
  37. id=3,
  38. color=[255, 128, 0],
  39. type='',
  40. swap='left_thumb1'),
  41. 4:
  42. dict(
  43. name='right_forefinger4',
  44. id=4,
  45. color=[255, 153, 255],
  46. type='',
  47. swap='left_forefinger4'),
  48. 5:
  49. dict(
  50. name='right_forefinger3',
  51. id=5,
  52. color=[255, 153, 255],
  53. type='',
  54. swap='left_forefinger3'),
  55. 6:
  56. dict(
  57. name='right_forefinger2',
  58. id=6,
  59. color=[255, 153, 255],
  60. type='',
  61. swap='left_forefinger2'),
  62. 7:
  63. dict(
  64. name='right_forefinger1',
  65. id=7,
  66. color=[255, 153, 255],
  67. type='',
  68. swap='left_forefinger1'),
  69. 8:
  70. dict(
  71. name='right_middle_finger4',
  72. id=8,
  73. color=[102, 178, 255],
  74. type='',
  75. swap='left_middle_finger4'),
  76. 9:
  77. dict(
  78. name='right_middle_finger3',
  79. id=9,
  80. color=[102, 178, 255],
  81. type='',
  82. swap='left_middle_finger3'),
  83. 10:
  84. dict(
  85. name='right_middle_finger2',
  86. id=10,
  87. color=[102, 178, 255],
  88. type='',
  89. swap='left_middle_finger2'),
  90. 11:
  91. dict(
  92. name='right_middle_finger1',
  93. id=11,
  94. color=[102, 178, 255],
  95. type='',
  96. swap='left_middle_finger1'),
  97. 12:
  98. dict(
  99. name='right_ring_finger4',
  100. id=12,
  101. color=[255, 51, 51],
  102. type='',
  103. swap='left_ring_finger4'),
  104. 13:
  105. dict(
  106. name='right_ring_finger3',
  107. id=13,
  108. color=[255, 51, 51],
  109. type='',
  110. swap='left_ring_finger3'),
  111. 14:
  112. dict(
  113. name='right_ring_finger2',
  114. id=14,
  115. color=[255, 51, 51],
  116. type='',
  117. swap='left_ring_finger2'),
  118. 15:
  119. dict(
  120. name='right_ring_finger1',
  121. id=15,
  122. color=[255, 51, 51],
  123. type='',
  124. swap='left_ring_finger1'),
  125. 16:
  126. dict(
  127. name='right_pinky_finger4',
  128. id=16,
  129. color=[0, 255, 0],
  130. type='',
  131. swap='left_pinky_finger4'),
  132. 17:
  133. dict(
  134. name='right_pinky_finger3',
  135. id=17,
  136. color=[0, 255, 0],
  137. type='',
  138. swap='left_pinky_finger3'),
  139. 18:
  140. dict(
  141. name='right_pinky_finger2',
  142. id=18,
  143. color=[0, 255, 0],
  144. type='',
  145. swap='left_pinky_finger2'),
  146. 19:
  147. dict(
  148. name='right_pinky_finger1',
  149. id=19,
  150. color=[0, 255, 0],
  151. type='',
  152. swap='left_pinky_finger1'),
  153. 20:
  154. dict(
  155. name='right_wrist',
  156. id=20,
  157. color=[255, 255, 255],
  158. type='',
  159. swap='left_wrist'),
  160. 21:
  161. dict(
  162. name='left_thumb4',
  163. id=21,
  164. color=[255, 128, 0],
  165. type='',
  166. swap='right_thumb4'),
  167. 22:
  168. dict(
  169. name='left_thumb3',
  170. id=22,
  171. color=[255, 128, 0],
  172. type='',
  173. swap='right_thumb3'),
  174. 23:
  175. dict(
  176. name='left_thumb2',
  177. id=23,
  178. color=[255, 128, 0],
  179. type='',
  180. swap='right_thumb2'),
  181. 24:
  182. dict(
  183. name='left_thumb1',
  184. id=24,
  185. color=[255, 128, 0],
  186. type='',
  187. swap='right_thumb1'),
  188. 25:
  189. dict(
  190. name='left_forefinger4',
  191. id=25,
  192. color=[255, 153, 255],
  193. type='',
  194. swap='right_forefinger4'),
  195. 26:
  196. dict(
  197. name='left_forefinger3',
  198. id=26,
  199. color=[255, 153, 255],
  200. type='',
  201. swap='right_forefinger3'),
  202. 27:
  203. dict(
  204. name='left_forefinger2',
  205. id=27,
  206. color=[255, 153, 255],
  207. type='',
  208. swap='right_forefinger2'),
  209. 28:
  210. dict(
  211. name='left_forefinger1',
  212. id=28,
  213. color=[255, 153, 255],
  214. type='',
  215. swap='right_forefinger1'),
  216. 29:
  217. dict(
  218. name='left_middle_finger4',
  219. id=29,
  220. color=[102, 178, 255],
  221. type='',
  222. swap='right_middle_finger4'),
  223. 30:
  224. dict(
  225. name='left_middle_finger3',
  226. id=30,
  227. color=[102, 178, 255],
  228. type='',
  229. swap='right_middle_finger3'),
  230. 31:
  231. dict(
  232. name='left_middle_finger2',
  233. id=31,
  234. color=[102, 178, 255],
  235. type='',
  236. swap='right_middle_finger2'),
  237. 32:
  238. dict(
  239. name='left_middle_finger1',
  240. id=32,
  241. color=[102, 178, 255],
  242. type='',
  243. swap='right_middle_finger1'),
  244. 33:
  245. dict(
  246. name='left_ring_finger4',
  247. id=33,
  248. color=[255, 51, 51],
  249. type='',
  250. swap='right_ring_finger4'),
  251. 34:
  252. dict(
  253. name='left_ring_finger3',
  254. id=34,
  255. color=[255, 51, 51],
  256. type='',
  257. swap='right_ring_finger3'),
  258. 35:
  259. dict(
  260. name='left_ring_finger2',
  261. id=35,
  262. color=[255, 51, 51],
  263. type='',
  264. swap='right_ring_finger2'),
  265. 36:
  266. dict(
  267. name='left_ring_finger1',
  268. id=36,
  269. color=[255, 51, 51],
  270. type='',
  271. swap='right_ring_finger1'),
  272. 37:
  273. dict(
  274. name='left_pinky_finger4',
  275. id=37,
  276. color=[0, 255, 0],
  277. type='',
  278. swap='right_pinky_finger4'),
  279. 38:
  280. dict(
  281. name='left_pinky_finger3',
  282. id=38,
  283. color=[0, 255, 0],
  284. type='',
  285. swap='right_pinky_finger3'),
  286. 39:
  287. dict(
  288. name='left_pinky_finger2',
  289. id=39,
  290. color=[0, 255, 0],
  291. type='',
  292. swap='right_pinky_finger2'),
  293. 40:
  294. dict(
  295. name='left_pinky_finger1',
  296. id=40,
  297. color=[0, 255, 0],
  298. type='',
  299. swap='right_pinky_finger1'),
  300. 41:
  301. dict(
  302. name='left_wrist',
  303. id=41,
  304. color=[255, 255, 255],
  305. type='',
  306. swap='right_wrist'),
  307. },
  308. skeleton_info={
  309. 0:
  310. dict(link=('right_wrist', 'right_thumb1'), id=0, color=[255, 128, 0]),
  311. 1:
  312. dict(link=('right_thumb1', 'right_thumb2'), id=1, color=[255, 128, 0]),
  313. 2:
  314. dict(link=('right_thumb2', 'right_thumb3'), id=2, color=[255, 128, 0]),
  315. 3:
  316. dict(link=('right_thumb3', 'right_thumb4'), id=3, color=[255, 128, 0]),
  317. 4:
  318. dict(
  319. link=('right_wrist', 'right_forefinger1'),
  320. id=4,
  321. color=[255, 153, 255]),
  322. 5:
  323. dict(
  324. link=('right_forefinger1', 'right_forefinger2'),
  325. id=5,
  326. color=[255, 153, 255]),
  327. 6:
  328. dict(
  329. link=('right_forefinger2', 'right_forefinger3'),
  330. id=6,
  331. color=[255, 153, 255]),
  332. 7:
  333. dict(
  334. link=('right_forefinger3', 'right_forefinger4'),
  335. id=7,
  336. color=[255, 153, 255]),
  337. 8:
  338. dict(
  339. link=('right_wrist', 'right_middle_finger1'),
  340. id=8,
  341. color=[102, 178, 255]),
  342. 9:
  343. dict(
  344. link=('right_middle_finger1', 'right_middle_finger2'),
  345. id=9,
  346. color=[102, 178, 255]),
  347. 10:
  348. dict(
  349. link=('right_middle_finger2', 'right_middle_finger3'),
  350. id=10,
  351. color=[102, 178, 255]),
  352. 11:
  353. dict(
  354. link=('right_middle_finger3', 'right_middle_finger4'),
  355. id=11,
  356. color=[102, 178, 255]),
  357. 12:
  358. dict(
  359. link=('right_wrist', 'right_ring_finger1'),
  360. id=12,
  361. color=[255, 51, 51]),
  362. 13:
  363. dict(
  364. link=('right_ring_finger1', 'right_ring_finger2'),
  365. id=13,
  366. color=[255, 51, 51]),
  367. 14:
  368. dict(
  369. link=('right_ring_finger2', 'right_ring_finger3'),
  370. id=14,
  371. color=[255, 51, 51]),
  372. 15:
  373. dict(
  374. link=('right_ring_finger3', 'right_ring_finger4'),
  375. id=15,
  376. color=[255, 51, 51]),
  377. 16:
  378. dict(
  379. link=('right_wrist', 'right_pinky_finger1'),
  380. id=16,
  381. color=[0, 255, 0]),
  382. 17:
  383. dict(
  384. link=('right_pinky_finger1', 'right_pinky_finger2'),
  385. id=17,
  386. color=[0, 255, 0]),
  387. 18:
  388. dict(
  389. link=('right_pinky_finger2', 'right_pinky_finger3'),
  390. id=18,
  391. color=[0, 255, 0]),
  392. 19:
  393. dict(
  394. link=('right_pinky_finger3', 'right_pinky_finger4'),
  395. id=19,
  396. color=[0, 255, 0]),
  397. 20:
  398. dict(link=('left_wrist', 'left_thumb1'), id=20, color=[255, 128, 0]),
  399. 21:
  400. dict(link=('left_thumb1', 'left_thumb2'), id=21, color=[255, 128, 0]),
  401. 22:
  402. dict(link=('left_thumb2', 'left_thumb3'), id=22, color=[255, 128, 0]),
  403. 23:
  404. dict(link=('left_thumb3', 'left_thumb4'), id=23, color=[255, 128, 0]),
  405. 24:
  406. dict(
  407. link=('left_wrist', 'left_forefinger1'),
  408. id=24,
  409. color=[255, 153, 255]),
  410. 25:
  411. dict(
  412. link=('left_forefinger1', 'left_forefinger2'),
  413. id=25,
  414. color=[255, 153, 255]),
  415. 26:
  416. dict(
  417. link=('left_forefinger2', 'left_forefinger3'),
  418. id=26,
  419. color=[255, 153, 255]),
  420. 27:
  421. dict(
  422. link=('left_forefinger3', 'left_forefinger4'),
  423. id=27,
  424. color=[255, 153, 255]),
  425. 28:
  426. dict(
  427. link=('left_wrist', 'left_middle_finger1'),
  428. id=28,
  429. color=[102, 178, 255]),
  430. 29:
  431. dict(
  432. link=('left_middle_finger1', 'left_middle_finger2'),
  433. id=29,
  434. color=[102, 178, 255]),
  435. 30:
  436. dict(
  437. link=('left_middle_finger2', 'left_middle_finger3'),
  438. id=30,
  439. color=[102, 178, 255]),
  440. 31:
  441. dict(
  442. link=('left_middle_finger3', 'left_middle_finger4'),
  443. id=31,
  444. color=[102, 178, 255]),
  445. 32:
  446. dict(
  447. link=('left_wrist', 'left_ring_finger1'),
  448. id=32,
  449. color=[255, 51, 51]),
  450. 33:
  451. dict(
  452. link=('left_ring_finger1', 'left_ring_finger2'),
  453. id=33,
  454. color=[255, 51, 51]),
  455. 34:
  456. dict(
  457. link=('left_ring_finger2', 'left_ring_finger3'),
  458. id=34,
  459. color=[255, 51, 51]),
  460. 35:
  461. dict(
  462. link=('left_ring_finger3', 'left_ring_finger4'),
  463. id=35,
  464. color=[255, 51, 51]),
  465. 36:
  466. dict(
  467. link=('left_wrist', 'left_pinky_finger1'),
  468. id=36,
  469. color=[0, 255, 0]),
  470. 37:
  471. dict(
  472. link=('left_pinky_finger1', 'left_pinky_finger2'),
  473. id=37,
  474. color=[0, 255, 0]),
  475. 38:
  476. dict(
  477. link=('left_pinky_finger2', 'left_pinky_finger3'),
  478. id=38,
  479. color=[0, 255, 0]),
  480. 39:
  481. dict(
  482. link=('left_pinky_finger3', 'left_pinky_finger4'),
  483. id=39,
  484. color=[0, 255, 0]),
  485. },
  486. joint_weights=[1.] * 42,
  487. sigmas=[])