search

mjq11302010044 / Real-CE

Real-CE: A Benchmark for Chinese-English Scene Text Image Super-resolution (ICCV2023)
66 stars 6 forks source link

How to inference? #7

Open yongzhuo opened 6 months ago

yongzhuo commented 6 months ago

How to inference? I have some confuse, while using basicsr/models/sr_model.py and the model of 'RRDB_RealCE.pth', the code:

print(img.shape)
output = self.net_g(img)
print(output.shape)
print:
torch.Size([1, 8120, 9792, 3])
torch.Size([1, 4, 39168, 12])

but the output shape is not a image, Is that right？Or any post-processing I need add?