41 lines
983 B
Python
41 lines
983 B
Python
import yaml
|
||
from enum import Enum
|
||
import os
|
||
BASE_DIR = os.environ['SAM_ANN_BASE_DIR']
|
||
|
||
DEFAULT_CONFIG_FILE = os.path.join(BASE_DIR, 'sam_ann', 'default.yaml')
|
||
CONFIG_FILE = os.path.join(BASE_DIR, 'sam_ann', 'isat.yaml')
|
||
CHECKPOINTS = os.path.join(BASE_DIR, 'checkpoints')
|
||
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
|
||
|
||
class CONTOURMode(Enum):
|
||
SAVE_MAX_ONLY = 0 # 只保留最多顶点的mask（一般为最大面积）
|
||
SAVE_EXTERNAL = 1 # 只保留外轮廓
|
||
SAVE_ALL = 2 # 保留所有轮廓 |