Skip to content

luisdiaz1997/BiomedicalNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

72 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Our Website


Built With


Train your own model (Tutorial)

FastChest - Tutorial using Jupyter Notebook to train using on Google's Colab. No need to install on your computer. You will need a Kaggle account to download the data.


Deploy on your computer

Instalation

Install anaconda 3 in your system https://www.anaconda.com/products/individual

Then create new environment

conda create --name fast python=3.6

Activate the environment

conda activate fast

Then install packages

conda install pytorch=1.4 torchvision -c pytorch
conda install -c fastai fastai
conda install flask gunicorn

Test run

python model.py sample/NORMAL2-IM-1427-0001.jpeg | python model.py sample/person1947_bacteria_4876.jpeg


Changing the paths

You can change the paths of where the model is located And where the predictions will be sent by changing the following in model.py

model_path = 'models/'
output_path = 'predictions/'

make sure that the path of the models always contains the file "dense"


Deployment

gunicorn --bind 0.0.0.0:5000 wsgi:app

Contributors

About

Pneumonia classifier using Dense121

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages