create tfrecords dir automatically (#10)

ethanyanjiali · May 22, 2020 · cbe6578 · cbe6578
1 parent 4365cdf
commit cbe6578
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Datasets/MPII/tfrecords_mpii.py b/Datasets/MPII/tfrecords_mpii.py
@@ -125,8 +125,8 @@ def parse_one_annotation(anno, image_dir):
 
 def main():
     print('Start to parse annotations.')
-    if not os.path.exists('./tfrecords_mpii'):
-        os.makedirs('./tfrecords_mpii')
+    if not os.path.exists('./tfrecords'):
+        os.makedirs('./tfrecords')
 
     with open('./mpii_human_pose_v1_u12_2/train.json') as train_json:
         train_annos = json.load(train_json)