This project aims to develop an AI model using PyTorch to classify lung X-ray images as either healthy or having pneumonia.
data/
: Contains the training, validation, and test datasets.src/
: Source code for data preprocessing, model definition, and training scripts.my-app
: front-end.requirements.txt
: List of dependencies.start.sh
: Start script.README.md
: Project documentation (here ^^).
- Install the dependencies:
pip install -r requirements.txt
- Run script
sh start.sh