Classification of chest x-ray images with pneumonia and normal using the kaggle dataset Chest X-Ray Images (Pneumonia), in which in this case tensorflow is used, the model is saved in .h5 and .tflite and the model is evaluated.
Google Colab is used to train the model where to download the dataset from Google Colab, as it is in Kaggle, you must first go to the configuration of the Kaggle account and click on the API "Create New Token" where a file "kaggle.json" will be downloaded. This file must be saved in a folder where you want to download the dataset.
To download the dataset copy the API command to be able to download it
Indicate the folder where the kaggle.json is located and download it with the api command of the dataset
loss and accuracy are observed
The confusion matrix is observed
The model is evaluated
- Use Dockerfile
- Use virtual enviroments and apply requirements.txt
pip install -r requirements.txt