semantic-pyramid-pytorch

semantic-pyramid-pytorch copied to clipboard

Pre-trained classifier

64

In the pre-trained classifier, why the output layer is not the softmax classifier for classifying 365 labels? Also, are the classifier layers also pre-trained in VGG classifier?

ayush1801

mask in reconstruction loss

1

I think your reconstruction loss is missing mask, which should be applied after subtracting two features as in (3) of the paper.  https://github.com/rosinality/semantic-pyramid-pytorch/blob/6d4796cbb2abdbe9afaafb8dae75b60a3b619d2d/train.py#L146

jychoi118

Could you please provide the pre-trained models?

KaiQiao1992