File size: 1,979 Bytes
ab9857f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
import DeepDeformationMapRegistration.utils.constants as C
import re
import os
class ConfigurationFile:
def __init__(self,
file_path: str):
self.__file = file_path
self.__load_configuration()
def __load_configuration(self):
fd = open(self.__file, 'r')
file_lines = fd.readlines()
for line in file_lines:
if '#' not in line and line != '\n':
match = re.match('(.*)=(.*)', line)
if match[1] in C.__dict__.keys():
# Careful with eval!!
try:
new_val = eval(match[2])
except NameError:
new_val = match[2]
old = C.__dict__[match[1]]
C.__dict__[match[1]] = new_val
# Special case
if match[1] == 'GPU_NUM':
C.__dict__[match[1]] = str(new_val)
os.environ['CUDA_VISIBLE_DEVICES'] = C.GPU_NUM
if match[1] == 'EPOCHS':
C.__dict__[match[1]] = new_val
C.__dict__['SAVE_EPOCH'] = new_val // 10
C.__dict__['VERBOSE_EPOCH'] = new_val // 10
if match[1] == 'SAVE_EPOCH' or match[1] == 'VERBOSE_EPOCH':
if new_val is not None:
C.__dict__[match[1]] = C.__dict__['EPOCHS'] // new_val
else:
C.__dict__[match[1]] = None
if match[1] == 'VALIDATION_ERR_LIMIT_COUNTER':
C.__dict__[match[1]] = new_val
C.__dict__['VALIDATION_ERR_LIMIT_COUNTER_BACKUP'] = new_val
print('INFO: Updating constant {}: {} -> {}'.format(match[1], old, C.__dict__[match[1]]))
else:
print('ERROR: Unknown constant {}'.format(match[1]))
|