diff --git a/core/data.py b/core/data.py index 08b46de..3c0d8e0 100644 --- 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))