Dataset: https://www.kaggle.com/datasets/paultimothymooney/chest-xray-pneumonia
train.py
yields the output below:
Moved 158 images to data\test\NORMAL
Moved 427 images to data\test\PNEUMONIA
epoch train_loss valid_loss accuracy time
0 0.494486 0.407573 0.880444 08:28
epoch train_loss valid_loss accuracy time
0 0.237182 0.165131 0.955594 09:52
Test on two random samples from test/NORMAL
and two random samples from test/PNEUMONIA
.
Both samples were excluded from the training set.
Image: test/NORMAL\NORMAL2-IM-1406-0001.jpeg
Prediction: NORMAL
Probabilities: tensor([0.7957, 0.2043])
Image: test/NORMAL\NORMAL2-IM-1326-0001.jpeg
Prediction: NORMAL
Probabilities: tensor([0.8174, 0.1826])
Image: test/PNEUMONIA\person136_bacteria_654.jpeg
Prediction: PNEUMONIA
Probabilities: tensor([0.0047, 0.9953])
Image: test/PNEUMONIA\person1910_bacteria_4814.jpeg
Prediction: PNEUMONIA
Probabilities: tensor([0.0144, 0.9856])