深哈希火炬
一些Deep Hash算法基线的实现。
怎么跑
我的环境是
python==3.7.0 torchvision==0.5.0 pytorch==1.4.0
您可以轻松地训练和测试任何算法
pyhon DSH.py
pyhon DPSH.py
pyhon DHN.py
pyhon DSDH.py
如果您有任何问题,请随时通过电子邮件( )与我联系或提出问题。
精确召回曲线
我在DSH.py中添加了一些代码:
if "cifar10-1" == config["dataset"] and epoch > 29:
P, R = pr_curve(trn_binary.numpy(), tst_binary.numpy(), trn_label.numpy(), tst_label.numpy())
print(f'Precision
1