Breast Cancer Detection using Tensorflow with GUI for testing with better user friendly.
- Use Google Colab for Training(Because of faster training and High RAM and GPU)
- Create folder named Breast Cancer Detection in Google Drive
- Inside Breast Cancer Detection Create 2 folders naming data and weights
- Inside Data folder create train and validation folders and upload all the images of train and validation into respective folders
- The Structure will look like this:
- upload the Breast-Cancer-Detection.ipynb file into drive or Google Colab
- Run the Breast-Cancer-Detection.ipynb file (Architecture is ResNet50 and inital weights are from imageNet)
- After running the whole file. Inside the weights folder the model file named BestDetection.h5 will be generated
- Tensorflow == 1.15 (if GPU is available install Tensorflow-gpu)
- OpenCv
- Numpy
- setuptools( >= 41.0.0)
- PyQt5
-
IF GPU is available:
- If NVIDIA Graphics card is available download and install CUDA and cuDNN.
- CUDA download link:
- cuDNN download link:
Follow currect steps to install CUDA and cuDNN(Make sure that both CUDA and cuDNN are of same version)
Tensorflow =
pip3 install tensorflow-gpu==1.15
- If NVIDIA Graphics card is available download and install CUDA and cuDNN.
-
Tensorflow =
pip3 install tensorflow==1.15
-
OpenCv =
pip3 install opencv-python
-
Numpy =
pip3 install numpy
-
setuptools =
pip3 install --upgrade setuptools
-
PyQt5 =
pip3 install PyQt5
- ipynb:
src\Breast-Cancer-Detection.ipynb
- python:
src\MainGUI.py
- UI :
src\MainGUI.ui