# Pneumonia_detection_Pytorch

Objective: To use Pytorch Resnet model to classify if the xray image of chest belongs to a patient suffering from pneumonia or not.

                

Model Architecture: 

model: resnet18

criterion = CrossEntropyLoss

optimizer = SGD(Stochastic gradient descent )

 

Result: