DeepHash-pytorch

收录理由

做图像检索或相似度搜索的团队，在选定某个哈希方法之前，往往需要在同一份数据上横向对比多种算法的效果。这个仓库把九种PyTorch实现的深度哈希算法集中在一起，DPSH、DSH、DHN、HashNet等都能通过单个脚本直接跑起来，README里还附了原论文和原始代码仓库的链接，方便你对照已发表的结果来验证实现是否可靠。仓库里也提供了CIFAR-10、NUS-WIDE、MirFlickr的数据切分脚本，ImageNet和COCO的下载地址也一并给出，另外带一个matplotlib演示脚本，可以直接看到检索输出的可视化效果。代码基于较旧的PyTorch 1.4.0和Python 3.7环境，实际使用时可能需要调整依赖版本，但作为哈希检索的基线参考，这个项目是个很顺手的起点。