The question about input of SP-Net

yccyenchicheng / pytorch-SegInpaint

A PyTorch Implementation of Segmentation for Image Inpainting based on SPG-Net: https://arxiv.org/abs/1805.03356

19 stars 2 forks source link

The question about input of SP-Net #9

Open super233 opened 2 years ago

super233 commented 2 years ago

Hi, thanks for your sharing code. I have noticed that the label map input of SP-Net is one hot, have you try to replace it with the softmax output of Semantic FCN?