In this repo, you will find a demonstration of how to build a Renal Health Classification model. The model is trained using Transfer Learning and the VGG16 architecture.
- AWS | AWS CLI
- DVC | DVC1.17
- Python | Python3.8
- MLFlow | MLFlow1.19
- TensorFlow | TensorFlow2.5
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
input_1 (InputLayer) [(None, 224, 224, 3)] 0
block1_conv1 (Conv2D) (None, 224, 224, 64) 1792
block1_conv2 (Conv2D) (None, 224, 224, 64) 36928
block1_pool (MaxPooling2D) (None, 112, 112, 64) 0
block2_conv1 (Conv2D) (None, 112, 112, 128) 73856
block2_conv2 (Conv2D) (None, 112, 112, 128) 147584
block2_pool (MaxPooling2D) (None, 56, 56, 128) 0
block3_conv1 (Conv2D) (None, 56, 56, 256) 295168
block3_conv2 (Conv2D) (None, 56, 56, 256) 590080
block3_conv3 (Conv2D) (None, 56, 56, 256) 590080
block3_pool (MaxPooling2D) (None, 28, 28, 256) 0
block4_conv1 (Conv2D) (None, 28, 28, 512) 1180160
block4_conv2 (Conv2D) (None, 28, 28, 512) 2359808
block4_conv3 (Conv2D) (None, 28, 28, 512) 2359808
block4_pool (MaxPooling2D) (None, 14, 14, 512) 0
block5_conv1 (Conv2D) (None, 14, 14, 512) 2359808
block5_conv2 (Conv2D) (None, 14, 14, 512) 2359808
block5_conv3 (Conv2D) (None, 14, 14, 512) 2359808
block5_pool (MaxPooling2D) (None, 7, 7, 512) 0
flatten (Flatten) (None, 25088) 0
dense (Dense) (None, 4) 100356
=================================================================
Total params: 14815044 (56.51 MB)
Trainable params: 100356 (392.02 KB)
Non-trainable params: 14714688 (56.13 MB)
_________________________________________________________________
(venv) prajwal@PK-2 Renal-Health % dvc dag
+----------------+ +--------------------+
| data_ingestion | | prepare_base_model |
+----------------+***** +--------------------+
* ***** *
* ****** *
* *** *
** +----------+
** | training |
*** +----------+
*** ***
** **
** **
+------------+
| evaluation |
+------------+
- Clone the repository
git clone https://github.com/prajwal3104/Renal-Health.git- Create a virtual environment
python3.8 -m venv venv- Activate the virtual environment
source venv/bin/activate- Install the requirements
pip install -r requirements.txt- Run the project or the pipeline's
python main.py- Website: Prajwaal.dev
- Twitter: Prajwal_b_k
- Github: prajwal3104
- LinkedIn: prajwal-kumbar
Contributions, issues and feature requests are welcome!
Feel free to check issues page.
Give a ⭐️ if this project helped you!