Fix Bugs in partseg.py

add mkdir os.path.join(DATA_DIR, 'shapenet_part_seg_hdf5_data')
tiangexiang · Aug 11, 2021 · 9c9b3e7 · 9c9b3e7
1 parent d213eb4
commit 9c9b3e7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/core/data.py b/core/data.py
@@ -41,6 +41,7 @@ def download_shapenetpart():
         www = 'https://shapenet.cs.stanford.edu/media/shapenet_part_seg_hdf5_data.zip'
         zipfile = os.path.basename(www)
         os.system('wget %s --no-check-certificate; unzip %s' % (www, zipfile))
+        os.mkdir(os.path.join(DATA_DIR, 'shapenet_part_seg_hdf5_data'))
         os.system('mv %s %s' % (zipfile[:-4], os.path.join(DATA_DIR, 'shapenet_part_seg_hdf5_data')))
         os.system('rm %s' % (zipfile))