frankkramer-lab / riadd.aucmedi

Multi-Disease Detection in Retinal Imaging based on Ensembling Heterogeneous Deep Learning Models
GNU General Public License v3.0
31 stars 10 forks source link

How can we combine all the models into a big model? #22

Open linlanSu opened 4 months ago

linlanSu commented 4 months ago

Hi, I'm kind of new to the field of deep learning, is there any way that we can get a whole model, more specifically, something like I can use output = model(image) to predict the label of it directly, instead of run all the .py file of different cnn separately (like below). python scripts/detector_DenseNet201.py python scripts/detector_EfficientNetB4.py python scripts/classifier_DenseNet201.py python scripts/classifier_InceptionV3.py python scripts/classifier_ResNet152.py python scripts/classifier_EfficientNetB4.py