Spaces:
Running
Running
Nguyễn Bá Thiêm
commited on
Commit
·
144016d
1
Parent(s):
1616e81
Remove unnecessary code in srflow.py
Browse files- models/SRFlow/srflow.py +0 -1
models/SRFlow/srflow.py
CHANGED
|
@@ -68,7 +68,6 @@ def test(lr, conf_path='models/SRFlow/code/confs/SRFlow_DF2K_4X.yml', heat=0.7):
|
|
| 68 |
sr = rgb(torch.clamp(sr_t, 0, 1))
|
| 69 |
sr = sr[:h * scale, :w * scale]
|
| 70 |
|
| 71 |
-
sr = Image.fromarray((sr).astype('uint8'))
|
| 72 |
return sr
|
| 73 |
|
| 74 |
if __name__ == '__main__':
|
|
|
|
| 68 |
sr = rgb(torch.clamp(sr_t, 0, 1))
|
| 69 |
sr = sr[:h * scale, :w * scale]
|
| 70 |
|
|
|
|
| 71 |
return sr
|
| 72 |
|
| 73 |
if __name__ == '__main__':
|