https://github.com/sartaj0/pytorchsegmentation
Providing a pipeline for training a segmentation model with different architectures
https://github.com/sartaj0/pytorchsegmentation
focal-loss pspnet pytorch segmentation tversky-loss unet
Last synced: about 2 months ago
JSON representation
Providing a pipeline for training a segmentation model with different architectures
- Host: GitHub
- URL: https://github.com/sartaj0/pytorchsegmentation
- Owner: sartaj0
- Created: 2021-05-24T13:01:29.000Z (almost 4 years ago)
- Default Branch: main
- Last Pushed: 2021-12-18T11:41:18.000Z (over 3 years ago)
- Last Synced: 2025-01-17T23:28:11.473Z (3 months ago)
- Topics: focal-loss, pspnet, pytorch, segmentation, tversky-loss, unet
- Language: Python
- Homepage: https://cv-project-deployed.herokuapp.com/
- Size: 19 MB
- Stars: 1
- Watchers: 2
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 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`
## Onnx Model
From [here](https://drive.google.com/drive/folders/15VQZVkfNaAGdoFVUabpdWid6aUw_i7d_?usp=sharing) you can download onnx converted model and use this [file](https://github.com/sartaj0/PytorchSegmentation/blob/main/inferenceOnnx.py) for inferencing with opencv.
## Output Example
| Image | Prediction Mask | Output |
| --- | --- | --- |
|
| 
| 
|
|
| 
| 
|
|
| 
| 
|
|
| 
| 
|