update

WJinhai · Apr 25, 2019 · 6eae16a · 6eae16a
1 parent 4857203
commit 6eae16a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/config.py b/config.py
@@ -3,7 +3,7 @@
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')  # sets device for model and PyTorch tensors
 
 # Model parameters
-im_size = 224
+im_size = 112
 channel = 3
 
 # Training parameters

diff --git a/extract.py b/extract.py
@@ -11,5 +11,5 @@ def extract(filename):
 
 
 if __name__ == "__main__":
-    if not os.path.isdir('data/cron20190326_resized'):
-        extract('data/cron20190326_resized.zip')
+    if not os.path.isdir('data/data/frame/cron20190326'):
+        extract('data/cron20190326.zip')
diff --git a/resize.py b/resize.py
@@ -3,11 +3,11 @@
 
 import cv2 as cv
 from tqdm import tqdm
-from config import im_size
 
+from config import im_size
 
-new_folder = r'D:\Users\foamliu\data\Image-Matching\data\cron20190326_resized'
-old_folder = r'D:\Users\foamliu\data\Image-Matching\data\data\frame\cron20190326'
+new_folder = 'data/cron20190326_resized'
+old_folder = 'data/data/frame/cron20190326'
 
 
 def resize_images(d):