Skip to content

Commit aeb66ef

Browse files
committed
rootnet path changed
1 parent 0e91952 commit aeb66ef

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

data/InterHand2.6M/dataset.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,10 @@ def __init__(self, transform, mode, annot_subset):
2828
self.annot_subset = annot_subset # all, human_annot, machine_annot
2929
self.img_path = '../data/InterHand2.6M/images'
3030
self.annot_path = '../data/InterHand2.6M/annotations'
31-
self.rootnet_output_path = '../data/InterHand2.6M/rootnet_output/rootnet_interhand2.6m_output_' + self.annot_subset + '_' + self.mode + '.json'
31+
if self.annot_subset == 'machine_annot' and self.mode == 'val':
32+
self.rootnet_output_path = '../data/InterHand2.6M/rootnet_output/rootnet_interhand2.6m_output_machine_annot_val.json'
33+
else:
34+
self.rootnet_output_path = '../data/InterHand2.6M/rootnet_output/rootnet_interhand2.6m_output_all_test.json'
3235
self.transform = transform
3336
self.joint_num = 21 # single hand
3437
self.root_joint_idx = {'right': 20, 'left': 41}

0 commit comments

Comments
 (0)