search

saurabheights / DiabeticRetinopathyDetection

Deep Learning Project on Diabetic Retinopathy Detection - TUM Team ID 47
GNU General Public License v3.0
21 stars 8 forks source link

about model #6

Closed mrzhangzizhen123 closed 5 years ago

mrzhangzizhen123 commented 5 years ago

I used the Resnet18 model ,but I don't how to used other model .please can you tell me how to use other models?such as Resnet34.Resnet50.The configuration file references this--https://github.com/saurabheights/DiabeticRetinopathyDetection/blob/master/src/transfer_learning_configuration.py

omsh commented 5 years ago

Change the parameter net_size in the configuration file you mentioned, please see here.

mrzhangzizhen123 commented 5 years ago

@omsh Thanks for your help. Do you have the code to test?In addition, have you debugged the visualization folder in the file?Can I see the debug code?