Skip to content

Latest commit

 

History

History
46 lines (39 loc) · 1.37 KB

README.md

File metadata and controls

46 lines (39 loc) · 1.37 KB

Zoidberg 2.0

Installation

  1. Clone the project and go into

    git clone git@github.com:EpitechMscPro2020/zoidberg2.0_2020_35.git
cd zoidberg2.0_2020_35

  2. Download the dataset and place chest_Xray folder at the repository root.

  3. Create python environment. You must have venv or conda on your machine

    # venv
python3 -m venv env
# or with conda
conda create -n env python=3

  4. Install dependencies

    source env/bin/activate
pip install --upgrade pip
pip install -r requirements.txt

  5. Launch jupyter web browser

    jupyter-notebook

Notebooks

  1. Dataset
    • Prepare data before training, it is mandatory before execution of another notebook
    • It will create a new file dataset.hdf5
  2. CNN
    • Train AI on previous created dataset with CNN model

Binary

We made a GUI to use our already trained model in ./model/cnn. You just need to upload image with upload button. The GUI will show the classification label and her prediction accuracy

Doesn't need notebooks execution or any dataset

You must have enabled the virtual env (goto installation 2.)

python prgm/ui/main.py