fix celebA dataset load (#66)

Author: hanjr92

tensorlayerx/files/dataset_loaders/celebA_dataset.py

@@ -27,19 +27,21 @@ def load_celebA_dataset(path='data'):
 
     data_dir = 'celebA'
     filename, drive_id = "img_align_celeba.zip", "0B7EVK8r0v71pZjFTYXZWM3FlRnM"
-    save_path = os.path.join(path, filename)
-    image_path = os.path.join(path, data_dir)
+    file_path = os.path.join(path, data_dir)
+    image_path = os.path.join(path, data_dir, "img_align_celeba")
+    save_path = os.path.join(path, data_dir, filename)
     if os.path.exists(image_path):
-        logging.info('[*] {} already exists'.format(save_path))
+        logging.info('[*] {} already exists'.format(image_path))
     else:
-        exists_or_mkdir(path)
-        download_file_from_google_drive(drive_id, save_path)
-        zip_dir = ''
+        if not os.path.exists(save_path):
+            exists_or_mkdir(file_path)
+            download_file_from_google_drive(drive_id, save_path)
+            zip_dir = ''
         with zipfile.ZipFile(save_path) as zf:
             zip_dir = zf.namelist()[0]
-            zf.extractall(path)
-        os.remove(save_path)
-        os.rename(os.path.join(path, zip_dir), image_path)
+            zf.extractall(file_path)
+        # os.remove(save_path)
+        # os.rename(os.path.join(path, zip_dir), image_path)
 
     data_files = load_file_list(path=image_path, regx='\\.jpg', printable=False)
     for i, _v in enumerate(data_files):