pytorch-Pyramid-Attention-Networks-PAN-

train code + inference code were released. [pytorch version]

PAN network：

FPA:

GAU:

How to train

first enter the directory: and follow the instruction: 1.write the dataset path in mypath.py 2. and run

python train.py

after trainning,it will save two model: convnet.pth,pan.pth

How to Inference：

#run:

python inference.py

input:

the Pic in validation dataset:

the Pic off the dataset:

output:

the Pic in validation dataset:

the Pic off the dataset: