fix bug of color model

f4a88e2e · LielinJiang · 8ee45431 · f4a88e2e
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

applications/DeOldify/predict.py applications/DeOldify/predict.py +1 -1

未找到文件。
--- a/applications/DeOldify/predict.py
+++ b/applications/DeOldify/predict.py
@@ -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)