diff --git a/applications/DeOldify/predict.py b/applications/DeOldify/predict.py index 9ac857f7b506e0ca3464be1ea2942f42005d582f..1f73fd45b499599bf5935e8d6aa85bd8a1fdddff 100644 --- 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)