fidler-lab / curve-gcn

Official PyTorch code for Curve-GCN (CVPR 2019)
GNU General Public License v3.0
841 stars 134 forks source link

Boundary Pred in Ablation Study. #15

Closed LetsGoFir closed 4 years ago

LetsGoFir commented 4 years ago

Hi there~ You said below in ablation study Boundary Pred adds the boundary prediction branch to our CNN. Does the 'Boundary Pred' mean below In order to help the model see image boundaries, we supervise two additional branches, i.e. an edge branch and a vertex branch, on top of the CNN encoder’s feature map F c , both of which consist of one 3 × 3 convolutional layer and one fully-connected layer. These branches are trained to predict the probability of existence of an object edge/vertex on a 28 × 28 grid. I am a little confused, because the GCN also predict the boundary.