import yaml from enum import Enum DEFAULT_CONFIG_FILE = 'default.yaml' CONFIG_FILE = 'isat.yaml' def load_config(file): with open(file, 'rb')as f: cfg = yaml.load(f.read(), Loader=yaml.FullLoader) return cfg def save_config(cfg, file): s = yaml.dump(cfg) with open(file, 'w') as f: f.write(s) return True class STATUSMode(Enum): VIEW = 0 CREATE = 1 EDIT = 2 class DRAWMode(Enum): POLYGON = 0 SEGMENTANYTHING = 1 class CLICKMode(Enum): POSITIVE = 0 NEGATIVE = 1 class MAPMode(Enum): LABEL = 0 SEMANTIC = 1 INSTANCE = 2