search

swuxyj / DeepHash-pytorch

Implementation of Some Deep Hash Algorithms, Including DPSH、DSH、DHN、HashNet、DSDH、DTSH、DFH、GreedyHash、CSQ.
MIT License
495 stars 116 forks source link

Greedy Hash如何在多标签数据NUS-WIDE, COCO上训练 #40

Open ssqiao opened 1 year ago

ssqiao commented 1 year ago

非常感谢你的project，实在太棒了。有个问题请教，Greedy Hash是基于交叉熵目标分类损失，但是NUS和COCO数据集是多标签图像，如何训练Greedy Hash，看到你给了结果，但不知道怎么训练？

swuxyj commented 1 year ago

也是用的torch.nn.CrossEntropyLoss()，这两个数据集用相同的流程也能跑出一个结果出来

ssqiao commented 1 year ago

应该是BCELoss吧，普通的CrossEntropy似乎不能处理多标签分类问题，只能处理单标签分类