Andy-zhujunwen / danet-pytorch

[pytorch] DANet: Dual Attention Network for Scene Segmentation
70 stars 24 forks source link

danet-pytorch

[pytorch] DANet: Dual Attention Network for Scene Segmentation

network:

image

dataset:

cityspaces

how to train:

write the dataset path in mypath.py and run

python train.py

after trainning,it will save a model: danet.pth

How to Inference:

run:

python inference.py

inference:

input:

image

output:

image