weidai00 commited on
Commit
f0ab20b
·
verified ·
1 Parent(s): a0dc447

Update AV/config/config_test_general.py

Browse files
Files changed (1) hide show
  1. AV/config/config_test_general.py +4 -1
AV/config/config_test_general.py CHANGED
@@ -107,11 +107,14 @@ def set_dataset(name):
107
  if dataset_name == 'DRIVE' or dataset_name == 'LES' or dataset_name == 'hrf':
108
  stride_height = 50
109
  stride_width = 50
110
- elif dataset_name == 'ukbb':
 
 
111
  stride_height = 150
112
  stride_width = 150
113
  else:
114
  use_CAM=True
 
115
  stride_height = 150
116
  stride_width = 150
117
 
 
107
  if dataset_name == 'DRIVE' or dataset_name == 'LES' or dataset_name == 'hrf':
108
  stride_height = 50
109
  stride_width = 50
110
+ elif dataset_name == 'ukbb' or dataset_name == 'hrf':
111
+ use_CAM=False
112
+ use_resize = True
113
  stride_height = 150
114
  stride_width = 150
115
  else:
116
  use_CAM=True
117
+ use_resize = True
118
  stride_height = 150
119
  stride_width = 150
120