Skip to content

includeamin/WBC-Classification

Repository files navigation

WBC-Classification

Classification of WBC ( White Blood Cells ) with CNN . (Convectional Neural Network)

Donate

HitCount

DATASET

Requirements

Keras==2.1.5
numpy==1.18.1
matplotlib==3.1.1
imutils==0.5.3
scikit_learn==0.23.1

Train and save trained model

Use this command to train the model and save model

git clone 
cd WBC-Classification
python3 learning.py -d ./CNN/datasets/TRAIN/ -m ./TrainedModel/model_epoch_100.hdf5

after train you will see result plot:

Test the model

python3 load_test_model.py -d ./CNN/datasets/TEST -m ./TrainedModel/model_epoch_100.hdf5

NOTE

default epoch count is 100. for change it, just edit cp variable in learning.py

Todo

  • more customizable