Skip to content

Latest commit

 

History

History
35 lines (21 loc) · 575 Bytes

README.md

File metadata and controls

35 lines (21 loc) · 575 Bytes

Cats Segmentation

Installing

python 3

if you have cuda 10.2 install as described below, else choose your specifications

pip install torch torchvision 
pip install -r requirements.txt

Usage

python predict.py --model-path path/to/model.pth
--image-path /path/to/image.jpg 
--out-path ~/path/to/out_image.jpg

Model

DeepLabV3+ model with a pre-trained ResNet-50 backbone

Download