dataset

Snailpong · Jan 12, 2021 · 3d90d8a · 3d90d8a
1 parent 972a185
commit 3d90d8a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 8 deletions.
diff --git a/filter_constant.py b/filter_constant.py
@@ -18,8 +18,8 @@
 Q_TOTAL = Q_ANGLE * Q_TENSOR
 FILTER_VOL = PATCH_SIZE ** 3
 
-TRAIN_DIV = 3
-SAMPLE_RATE = 3
+TRAIN_DIV = 5
+SAMPLE_RATE = 1
 SHARPEN = 'False'
 BLEND_THRESHOLD = 10
 

diff --git a/train.py b/train.py
@@ -114,7 +114,7 @@ def train_qv(im_LR, im_HR, w, kmeans, std, Q, V, count):
  file_name = file.split('\\')[-1].split('.')[0]
  filestart = time.time()
 
- if file_idx >= 100:
+ if file_idx >= 30:
  break
 
  if file in finished_files:

diff --git a/util.py b/util.py
@@ -13,11 +13,10 @@
 def make_dataset(dir):
  images = []
  assert os.path.isdir(dir), '%s is not a valid directory' % dir
- for root, _, fnames in sorted(os.walk(dir)):
- for fname in fnames:
- if fname.endswith('.nii.gz'):
- path = os.path.join(root, fname)
- images.append(path)
+ for fname in sorted(os.listdir(dir)):
+ if fname.endswith('.nii.gz'):
+ path = os.path.join(dir, fname)
+ images.append(path)
  return images