ResNet-18 using PyTorch on Chest X-ray dataset
-
Implementation of ResNet-18 from scratch
-
Using a pre-trained ResNet-18
a) Using the pre-trained network for classification by retraining only the newly added classifier part
b) Fine-tuning the pre-trained ResNet-18
-
Modifying the ResNet-18 model architecture
a) Modifying the original ResNet-18 into ResNet-14 and training from scratch
b) Modifying the pre-trained ResNet-18 and fine-tuning
Dataset:- https://www.kaggle.com/datasets/tolgadincer/labeled-chest-xray-images
Pre-trained ResNet-18:- https://pytorch.org/vision/main/models/generated/torchvision.models.resnet18.html