Project Title

Detect cells in microfluidic channels using YOLO

Installation

pip setup.py .

Edit the configuration config.yaml
Ensure that GPU's are recognized and used for training by installing the right CUDA version and loading the GPU's training.py
Start the training script by executing training.py

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
data/training_data		data/training_data
README.md		README.md
UNET_model.py		UNET_model.py
augmentation.py		augmentation.py
config.yaml		config.yaml
helper_functions.py		helper_functions.py
loss.py		loss.py
predict.py		predict.py
predict_full_images.py		predict_full_images.py
setup.py		setup.py
training.py		training.py
weights.h5		weights.h5