Keras==2.1.5
numpy==1.18.1
matplotlib==3.1.1
imutils==0.5.3
scikit_learn==0.23.1
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:
python3 load_test_model.py -d ./CNN/datasets/TEST -m ./TrainedModel/model_epoch_100.hdf5
default epoch count is 100. for change it, just edit cp
variable in learning.py
- more customizable