Skip to content

Commit 4f25b95

Browse files
committed
Initial commit
1 parent dc24f8f commit 4f25b95

4 files changed

Lines changed: 98 additions & 1 deletion

File tree

README.md

Lines changed: 70 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,71 @@
11
# Efficient-Segmentation-Networks
2-
Lightweight models for real-time semantic segmentationon PyTorch (include SQNet, LinkNet, SegNet, UNet, ENet, ERFNet, EDANet, ESPNet, LEDNet, ESNet, CGNet, DABNet, Fast-SCNN, ContextNet, etc.)
2+
[![python-image]][python-url]
3+
[![pytorch-image]][pytorch-url]
4+
[![lic-image]][lic-url]
5+
6+
This project aims at providing a easy-to-use, modifiable reference implementation for real-time semantic segmentation models using PyTorch.
7+
8+
<p align="center"><img width="100%" src="docs/image-1.png" /></p>
9+
10+
## Support
11+
12+
#### Model
13+
- [SQNet](https://openreview.net/pdf?id=S1uHiFyyg)
14+
- [LinkNet](https://arxiv.org/pdf/1707.03718.pdf)
15+
- [SegNet](https://arxiv.org/pdf/1511.00561.pdf)
16+
- [UNet](https://arxiv.org/pdf/1505.04597.pdf)
17+
- [ENet](https://arxiv.org/pdf/1606.02147.pdf)
18+
- [ERFNet](http://www.robesafe.uah.es/personal/eduardo.romera/pdfs/Romera17iv.pdf)
19+
- [EDANet](https://arxiv.org/ftp/arxiv/papers/1809/1809.06323.pdf)
20+
- [ESPNet](https://arxiv.org/pdf/1803.06815v2.pdf)
21+
- [LEDNet](https://arxiv.org/pdf/1905.02423v3.pdf)
22+
- [ESNet](https://arxiv.org/pdf/1906.09826v1.pdf)
23+
- [CCNet](https://arxiv.org/pdf/1811.08201.pdf)
24+
- [DABNet](https://arxiv.org/pdf/1907.11357.pdf)
25+
- [Fast-SCNN](https://arxiv.org/pdf/1902.04502.pdf)
26+
- [ContextNet](https://arxiv.org/pdf/1805.04554.pdf)
27+
28+
29+
30+
31+
32+
33+
34+
35+
36+
37+
38+
39+
40+
41+
42+
43+
44+
45+
46+
47+
48+
49+
50+
51+
52+
[python-image]: https://img.shields.io/badge/Python-3.x-ff69b4.svg
53+
[python-url]: https://www.python.org/
54+
[pytorch-image]: https://img.shields.io/badge/PyTorch-1.1-2BAF2B.svg
55+
[pytorch-url]: https://pytorch.org/
56+
[lic-image]: https://img.shields.io/aur/license/pac.svg
57+
[lic-url]: https://github.com/xiaoyufenfei/Efficient-Segmentation-Networks
58+
/blob/master/LICENSE
59+
60+
61+
62+
63+
64+
65+
66+
67+
68+
69+
70+
71+

docs/README.md

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
### Model & Backbone
2+
3+
| Model | Scratch | ResNet18 | ResNet50 | ResNet101 |
4+
| :--------: | :-----: | :------: | :------: | :-------: |
5+
| SQNet || | | |
6+
| LinkNet || | | |
7+
| SegNet || | | |
8+
| UNet || | | |
9+
| ENet || | | |
10+
| ERFNet || | | |
11+
| EDANet || | | |
12+
| ESPNet || | | |
13+
| LEDNet || | | |
14+
| ESNet || | | |
15+
| CGNet || | | |
16+
| DABNet || | | |
17+
| Fast-SCNN || | | |
18+
| ContextNet || | | |
19+
| CANet || | | |

docs/image-1.png

1.06 MB
Loading

docs/requirements.yml

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
name: seg_requirements
2+
dependencies:
3+
- python3
4+
- cuda 9.0
5+
- pip:
6+
- Image
7+
- tqdm
8+
- torch 1.1.0
9+
- torchvision 0.3.0

0 commit comments

Comments
 (0)