修复
Browse files- cyclegan.py +1 -0
cyclegan.py
CHANGED
|
@@ -100,6 +100,7 @@ class CYCLEGAN(object):
|
|
| 100 |
|
| 101 |
|
| 102 |
image = postprocess_output(pr)
|
|
|
|
| 103 |
image = Image.fromarray(np.uint8(image))
|
| 104 |
|
| 105 |
return image
|
|
|
|
| 100 |
|
| 101 |
|
| 102 |
image = postprocess_output(pr)
|
| 103 |
+
image = np.clip(image, 0, 255)
|
| 104 |
image = Image.fromarray(np.uint8(image))
|
| 105 |
|
| 106 |
return image
|