search

abhimanyudubey / confusion

Code for the ECCV 2018 paper "Pairwise Confusion for Fine-Grained Visual Classification"
https://arxiv.org/abs/1705.08016
200 stars 39 forks source link

pytorch loss #8

Closed twmht closed 5 years ago

twmht commented 5 years ago

Hi,

there are two losses (PairwiseConfusion and EntropicConfusion) defined in pytorch (https://github.com/abhimanyudubey/confusion/blob/master/confusion_pytorch/__init__.py), but the names confuse me.

which one should I use?

from the paper, EQ(5) is Euclidean Confusion, is this refer to PairwiseConfusion?

Where can I find the entropy confusion from the paper?

abhimanyudubey commented 5 years ago

Hi, the entropic confusion is described in this paper: http://papers.nips.cc/paper/7344-maximum-entropy-fine-grained-classification. You can try out both, hopefully, they should both provide improvements.