Moved image normalization into onnx models.
This commit is contained in:
@ -43,16 +43,9 @@ class SimCC(BaseModel):
|
||||
self.dy = 0
|
||||
self.scale = 0
|
||||
|
||||
norm_mean = -1 * np.array([123.675, 116.28, 103.53])
|
||||
norm_std = 1.0 / np.array([58.395, 57.12, 57.375])
|
||||
self.norm_mean = np.reshape(norm_mean, (1, 1, 3)).astype(np.float32)
|
||||
self.norm_std = np.reshape(norm_std, (1, 1, 3)).astype(np.float32)
|
||||
|
||||
def preprocess(self, image: np.ndarray):
|
||||
tensor, self.dx, self.dy, self.scale = image, 0, 0, 1
|
||||
tensor = tensor.astype(np.float32, copy=False)
|
||||
tensor += self.norm_mean
|
||||
tensor *= self.norm_std
|
||||
tensor = np.expand_dims(tensor, axis=0).transpose((0, 3, 1, 2))
|
||||
return tensor
|
||||
|
||||
|
||||
Reference in New Issue
Block a user