Skip to content

Creating a good Convolutional Neural Network to classify our data. I used Model Checking to save the best model and VGG16 and ResNet50 to leverage pre-trained models. I ultimately used pre-trained models to fine tune a model.

Notifications You must be signed in to change notification settings

AmirHHasani/Manual-tuning-of-Convolutional-Neural-Networks-for-a-medical-image-dataset-x-ray-images-

Repository files navigation

Manual tuning of Convolutional Neural Networks for a medical image dataset(x-ray images)

Note that model checking based on validation loss has been used to save the best model in the training process. Metrics such as True Negatives or True positives have been used from time to time.

In steps 1 to 4 we try to find a good structure and in step 5 (which has several steps of its own), tuning of the parameters happen.

Step 1 : Trying random CNN models on our dataset to find a good base model using Keras

Step 2 : Using the best model from Step 1 and adding data augmentation, dropout layer. Then using VGG16 and Imagenet weights in fine tuning a model and trying to get better results. I only unfroze the last 4 layers of the conv base(VGG16) to finetune at the end.

Step 3 : Using ResNet50 instead of VGG16 in the fine tuning process.

Step 4 : The best model so far.

Step 5 : Tuning different settings of our best model

About

Creating a good Convolutional Neural Network to classify our data. I used Model Checking to save the best model and VGG16 and ResNet50 to leverage pre-trained models. I ultimately used pre-trained models to fine tune a model.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published