CaitinZhao/cvpr2019_Pyramid-Feature-Attention-Network-for-Saliency-detection

Problem with inference

ojasve opened this issue · 2 comments

ojasve commented

There is a shape issue when running the inference with the vgg model provide. It says it accepts only int but float provided. Any ideas on how to run inference for the model provided?

gden138 commented

attention.py file row 59,modify :::: attention = Dense(int(C / 4), activation='relu')(attention)

  • 👍3
sairajk commented

Hi @ojasve and @gden138 ,

You can check my PyTorch implementation of this paper here.