search

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

How to change another detector network #20

Closed JJY0710 closed 3 years ago

JJY0710 commented 3 years ago

I want to change the FRCNN_model to another model, such as Casecade RCNN or other detection models trained by me.How can I do. Thank you.

Jakaria08 commented 3 years ago

In the following segment, I used FRCNN. You can add your own model here: https://github.com/Jakaria08/EESRGAN/blob/c646296f68c1158c62e6bf9eca5fc82103e4f120/model/ESRGAN_EESN_FRCNN_Model.py#L40

JJY0710 commented 3 years ago

In the following segment, I used FRCNN. You can add your own model here:

https://github.com/Jakaria08/EESRGAN/blob/c646296f68c1158c62e6bf9eca5fc82103e4f120/model/ESRGAN_EESN_FRCNN_Model.py#L40

OK, thank you very much