Spaces:
Running
Running
Update AV/config/config_test_general.py
Browse files
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 |
|