DANet_baseline

Environment

• Ubuntu 16.04.6 LTS
• Python 3.6
• cuda10.1+cudnn7.6.3

Requirement

• 安装 nvidia-apex , 并确保在code目录下
• pip install requirements.txt

Usage

将数据集文件夹ps_battles_orisize放到和train.sh同级目录下

Test

bash run.sh

Train

bash train.sh

请自行在./config的config文件中修改配置参数 将train_path,val_path,test_dir修改为自己的路径