notebooks for training the models
datasets for training the models
Datasets currently at: https://drive.google.com/drive/folders/1-qBEfke0XmfN13_zaccMZa5YH0vfvNJa?usp=drive_link
pretrained models just for testing
Trained models currently at: https://drive.google.com/drive/folders/1-qBEfke0XmfN13_zaccMZa5YH0vfvNJa?usp=drive_link
the apps/scripts for testing the models
Example images for testing app
simple web server for testing the models from anywhere without need to install the environment
Note About the dependencies, tensorflow 2.14.1 works with cuda 11 if you have cuda 12 you need to install tensorflow 2.15.X also you need to take care of tensorrt to make sure is compatible with cudnn
This folder constains Pipenv file for installing the dependencies. To install the dependencies, run the following command:
pip install pipenv
pipenv install
To run the app, run the following command:
pipenv run python testApp/test_app.py
This folder constains Pipenv file for installing the dependencies. To install the dependencies, run the following command:
pip install pipenv
pipenv install
Configure your jupyter to use the pipenv kernel and then run the notebooks in the notebooks folder. Follow the instructions in the notebooks to train the models.