Instance segmentation with U-Net/Mask R-CNN workflow using Keras & Ray Tune
(Instance segmentation with U-Net/Mask R-CNN workflow using Keras & Ray Tune)
Lung instance segmentation workflow uses Chest X-ray for predicting lung masks from the images using U-Net model.
git clone https://github.com/pegasus-isi/lung-instance-segmentation-workflow.git
python3 workflow.py --lung-img-dir inputs/train_images --lung-mask-img-dir inputs/train_masks
To Run the workflow using the production dataset, you must first obtain it.
pip3 install kaggle
python3 get-dataset.py
/lizard/projects/ml-workflows/lung-segmentation-workflow/lung-segmentation-data.tar.gz
and has a sha256
hash of f30a3d450dce65a4c0f93c9e408e0cd457023d70db9599032179c36758fbf5fc
python3 workflow.py
./data
which was created by get-dataset.py
pip3 -r requirements.txt
to install the required packageslung-instance-segmentation-workflow
directory and make a directory called output
python get-datatset.py
end-to-end.sh
script