关于对比算法的实现

Yunfan-Li / Contrastive-Clustering

Code for the paper "Contrastive Clustering" (AAAI 2021)

MIT License

289 stars 92 forks source link

关于对比算法的实现 #26

Closed zbw0329 closed 2 years ago

zbw0329 commented 2 years ago

您好，我对您CC的研究非常感兴趣。我想问一下您论文中实验部分，您的对比算法K-means和SC是如何实现的？能否简要介绍一下您对比算法都是如何实现的？我想要在自己的数据集上复现您的对比实验

Yunfan-Li commented 2 years ago

您好，论文中对比实验的结果都是从之前的工作PICA中直接照搬过来的，我们并没有跑这些baseline。

zbw0329 commented 2 years ago

您好，我想问一下imagenet-10和imagenet-dogs下的目录应该是怎样排布的？您在代码中dataset文件夹中介绍的那些tar包下载好后放在比如imagenet-10下的train文件夹中对吗？那test和val文件夹下需要放什么数据呢？我看见tiny-imagenet里面有三个文件夹

Yunfan-Li commented 2 years ago

您好，直接使用Pytorch提供的ImageFolder类就可以了，具体地可以参考 #10

zbw0329 commented 2 years ago

您好，我对cc进行了一些修改，添加了一个投影头，然后在imagenet-10上训练，在使用cluster.py进行评估的时候得到了如下的报错 1638888975(1) 在训练过程中没有报错，在用cluster.py的时候报错了，请问您知道什么原因可能导致这个报错吗？

Yunfan-Li commented 2 years ago

应该是目标聚类个数和ground truth不一致，您可能需要尝试修改ACC的计算方式。

zbw0329 commented 2 years ago

我发现是因为我读的label全是0，所以报错，还在排查是为什么，感谢您的帮助

zbw0329 commented 2 years ago

解决了，原因是我在放数据集的时候多加了一层文件夹train,导致在运行clusetr.py读不到数据进行验证，所以读的标签全是0 感谢您的研究工作以及您提供的帮助