base.py

import psutil
from pathlib import Path
import os
from utils import utils
from utils.utils import Singleton,read_yaml_file

# 读取模型yaml文件
class FileConfig(metaclass=Singleton):
    def __init__(self):
#         read model.yaml文件
        cur_path = Path(os.path.abspath(__file__))
        base_dir = cur_path.parent
        self.model_config_path = base_dir / "model.yaml"
        self.model_file = read_yaml_file(self.model_config_path.as_posix())
        # self.model_path = base_dir / "resnet152-b121ed2d.pth"
        # self.model_path = base_dir / "resnet18-5c106cde.pth"
        # self.model_path = base_dir / "resnet34-333f7ec4.pth"
        self.model_path = base_dir / "resnet101-5d3b4d8f.pth"

class BaseUI(object):
    def __init__(self):
        cur_path = Path(os.path.abspath(__file__))
        cur_dir = cur_path.parent
        self.file_init()
        self.model_init()
    def file_init(self):
    #     初始化读取yaml文件
        file_config_singleton = FileConfig()
        self.mdoel_config_path = file_config_singleton.model_config_path
        self.model_file = file_config_singleton.model_file
        self.model_path = file_config_singleton.model_path
    def model_init(self):
        self.mdoel_name = self.model_file['model_name']
        self.contrast_path = ('.' + str(self.model_file['contrast_path']))
        self.transforms_Resize = [int(item) for item in self.model_file['transform']['Resize']]
        self.transforms_Normalize_mean = [float(item) for item in self.model_file['transform']['Normalize']['mean']]
        self.transforms_Normalize_std = [float(item) for item in self.model_file['transform']['Normalize']['std']]
        self.transforms_centercrop_isExist = bool(self.model_file['transform']['centercrop']['isExist'])
        if self.transforms_centercrop_isExist:
            self.transforms_centercrop_value = int(self.model_file['transform']['centercrop']['value'])
baseui = BaseUI()