search

XPixelGroup / RDSR

(CVPR2022) Reflash Dropout in Image Super-Resolution
80 stars 7 forks source link

GENERALIZATION #3

Open nanmehta opened 2 years ago

nanmehta commented 2 years ago

CAN YOU PLEASE TELL ME HOW CAN WE CHECK THE CSM FOR OUR TRAINED MODEL? where do WE need TO DO ANY CHANGES?

nanmehta commented 2 years ago

hii, please help

liuxinaDaisy commented 2 years ago

I'm sorry I didn't see your question in time. For your demand there are three changes you need to make. 'init.py' in folder 'ModelZoo', folder 'NN' in folder 'ModelZoo', and variable 'model_list' in 'generate_map.py'.

nanmehta commented 2 years ago

okay thanks, but i have one doubt as to what does bold line 19-22 imply in model_list=[ ['RealSRResNetx4_details_8class_abs', 'ModelZoo/pretrained_models/RealSRResNetx4_details.pth']? I understand its a model name, but i could not find a model with this name in init.py. ['RealSRResNetx4_details_droplast_channel07_8class_abs', 'ModelZoo/pretrained_models/RealSRResNetx4_details_droplast_channel07.pth'], ] Is it necessary to define this line?

liuxinaDaisy commented 2 years ago

In 'generate_map.py', follow the variable 'model_list'.