Semantic Segmentation Library

Install

Clone github repo and install the necessary requirements

git clone https://github.com/trainyolo/segment
cd segment
pip install -r requirements.txt

Train

python train.py --dataset_path /path/to/dataset

Predict

python predict.py --source input.jpg   --weights /path/to/best_miou_model.pt
                           path/
                           path/*.jpg