PytorchSegmentation

Dependency

pytorch 1.8.1, CUDA 10.2
pip3 install torch==1.8.1+cu102 torchvision==0.9.1+cu102
Requirements: Pillow, opencv-python, tqdm, matplotlib
pip3 install Pillow opencv-python tqdm matplotlib

From here you can download onnx converted model and use this file for inferencing with opencv.

Name		Name	Last commit message	Last commit date
Latest commit History 27 Commits
images		images
model		model
modules		modules
.gitignore		.gitignore
README.md		README.md
data.json		data.json
inferenceOnnx.py		inferenceOnnx.py
requirements.txt		requirements.txt
train.py		train.py