提交 f4a88e2e 编写于 作者: L LielinJiang

fix bug of color model

上级 8ee45431
......@@ -74,7 +74,7 @@ class DeOldifyPredictor():
img += img_mean
img = img.transpose((1, 2, 0))
return (img * 255).astype('uint8')
return (img * 255).clip(0, 255).astype('uint8')
def post_process(self, raw_color, orig):
color_np = np.asarray(raw_color)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册