Jakaria08 / EESRGAN

Small-Object Detection in Remote Sensing (satellite) Images with End-to-End Edge-Enhanced GAN and Object Detector Network
GNU General Public License v3.0
279 stars 70 forks source link

Can we train to only get Super-Res images? #15

Closed purijs closed 3 years ago

purijs commented 3 years ago

Hi

I'm trying to make this work on my custom dataset, I only have jpegs but no annotation data and am only interested getting the SR images as output. Will the code work without annotations, where all can I modify the code?

Jakaria08 commented 3 years ago

There are many hardcoded parts in my code; sorry for that. Many parts are needed to be changed. Try to run this project: https://github.com/kuihua/EEGAN, it works only with superresolution.

For my project, you should use this model: https://github.com/Jakaria08/EESRGAN/blob/c646296f68c1158c62e6bf9eca5fc82103e4f120/model/ESRGAN_EESN_Model.py#L132 The dataset script also needed to be changed. Besides these you need to change some other code segments that are dependent.