utils.py 文件源码

def __init__(self, model, scale=1.050):
        self.input_size = model.input_size
        self.input_space = model.input_space
        self.input_range = model.input_range
        self.mean = model.mean
        self.std = model.std
        self.scale = scale
        self.tf = transforms.Compose([
            transforms.Scale(int(round(max(self.input_size)*self.scale))),
            transforms.CenterCrop(max(self.input_size)),
            transforms.ToTensor(),
            ToSpaceBGR(self.input_space=='BGR'),
            ToRange255(max(self.input_range)==255),
            transforms.Normalize(mean=self.mean, std=self.std)
        ])