读取BYOL原论文之后,找了一篇认可度较高的用pytorch实现的代码,自己训练1000次,利用得到的图像特征去做k-means聚类,聚类准确度很低,只有27%左右。cifar10数据集一般可以做到80%以上,这个结果有点低了。问一下大家有没有用这个代码训练一个准确度较高的模型,不同代码也可以,用于深度图像聚类。
代码:https://github.com/lucidrains/byol-pytorch
训练代码
模型评估