SegmentationPipeline -- Pytorch

A simple pytorch pipeline for semantic segmentation.

Requirements :

torch>=1.9.0
tqdm
albumentations>=1.0.3
opencv-python
Pillow>=8.3.1
numpy>=1.19.3
ttach
pandas

Use :

pip install -r requirements.txt

to install the required modules.