Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "a03d13c83d402a6c13866a756ebfaceb43b8e8b6"
Unverified Commit 5fb98229 authored by Maksim Skobeev's avatar Maksim Skobeev Committed by GitHub
Browse files

Update model.py

parent 601b568f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
RealESRGAN/model.py
View file @ 5fb98229
...@@ -75,7 +75,7 @@ class RealESRGAN: ...@@ -75,7 +75,7 @@ class RealESRGAN:
for i in range(batch_size, img.shape[0], batch_size): for i in range(batch_size, img.shape[0], batch_size):
res = torch.cat((res, self.model(img[i:i+batch_size])), 0) res = torch.cat((res, self.model(img[i:i+batch_size])), 0)
sr_image = res.permute((0,2,3,1)).clamp_(0, 1).cpu() sr_image = res.permute((0,2,3,1)).cpu().clamp_(0, 1)
np_sr_image = sr_image.numpy() np_sr_image = sr_image.numpy()
padded_size_scaled = tuple(np.multiply(p_shape[0:2], scale)) + (3,) padded_size_scaled = tuple(np.multiply(p_shape[0:2], scale)) + (3,)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment