Changed for ActEV inferencing

JunweiLiang · web-flow · commit 9a0d0c5f4cca · 2019-07-01T19:43:54.000-07:00
diff --git a/lib/dataset/JointsDataset.py b/lib/dataset/JointsDataset.py
@@ -192,7 +192,8 @@ def __getitem__(self, idx):
             'center': c,
             'scale': s,
             'rotation': r,
-            'score': score
+            'score': score,
+            'pid': db_rec['pid']  # added by Junwei. Pass the personid so we could output the kp results
         }
 
         return input, target, target_weight, meta

Original file line number	Diff line number	Diff line change
`@@ -192,7 +192,8 @@ def __getitem__(self, idx):`
`192`	`192`	`'center': c,`
`193`	`193`	`'scale': s,`
`194`	`194`	`'rotation': r,`
`195`		`- 'score': score`
	`195`	`+ 'score': score,`
	`196`	`+ 'pid': db_rec['pid'] # added by Junwei. Pass the personid so we could output the kp results`
`196`	`197`	`}`
`197`	`198`
`198`	`199`	`return input, target, target_weight, meta`