search

Lornatang / SRGAN-PyTorch

A simple and complete implementation of super-resolution paper.
Apache License 2.0
411 stars 105 forks source link

Errors when SR near to white colors #41

Closed SirSykon closed 2 years ago

SirSykon commented 2 years ago

Hello,

I've applied your model to images containing white and yellow sections. The SR results shows green instead of yellow but the strangest is the white color is replaced by a patron I suspect is provoked by storing NaN or something like that. I attach two examples. With the provided weights:

white_p yellow_p

After fine tuning with images from the same dataset: white yellow

Any ideas how to fix it?

Lornatang commented 2 years ago

I'll see the error, wait a minute

SirSykon commented 2 years ago

Any update? I'm intrigued about why the model fails with that kind of images. :)

Lornatang commented 2 years ago

I am sorry.~ Input and output range and RGB or BGR?

SirSykon commented 2 years ago

RGB. Should it be BGR?

Lornatang commented 2 years ago

no, RGB。

Lornatang commented 2 years ago

In the near future, I will update the model weights, you can try again later.

Lornatang commented 2 years ago

@SirSykon Try to pull the latest code.