/danet-pytorch

[pytorch] DANet: Dual Attention Network for Scene Segmentation

Primary LanguagePython

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