search

ZhiyuanDang / NNM

The PyTorch official implementation of the CVPR2021 Poster Paper NNM: Nearest Neighbor Matching for Deep Clustering.
Other
59 stars 7 forks source link

about visualization #10

Open haoweiclouds1 opened 2 years ago

haoweiclouds1 commented 2 years ago

作者大大您好，想请教一下你们文中Fig4. Confusion matrices和Fig5. 是如何出图的 感谢解答 能否提供参考代码 网络上没找到

再次感谢

ZhiyuanDang commented 2 years ago

你好，图4的混淆矩阵是通过eval.py的第86-88行的实现的，有参数可以选择混淆矩阵的保存位置。

至于图5，同样是在eval.py，我们可以获得每类的前多个原型，然后我按照他们的预测类标，手动将他们拼齐为一类并手动标注相似的真实标签。