文件名称:PCL_Sinkhorn
文件大小:581KB
文件格式:ZIP
更新时间:2024-05-08 00:48:22
Python
基于集群的自我监督学习方法的实现与评估 这种PyTorch实施是一个自我监督的学习网络,结合了基于比较学习和聚类学习。 K均值聚类部分被Sinkhorn-Knopp算法取代,产生了三个想法。 要求: 数据集 ImageNet,迷你100 CIFAR-10,CIFAR-100 的Python 3.8.3 PyTorch 1.7.1 CUDA 11.0 点安装tqdm 点安装tensorboard_logger 资料集: 训练和测试是根据类的文件夹生成的。 影像网 将图像打包到文件夹 mkdir val && mv ILSVRC2012_img_val.tar val/ && cd val && tar -xvf ILSVRC2012_img_val.tar wget -qO-https://raw.githubusercontent.com/soumith/imagene
【文件预览】:
PCL_Sinkhorn-main
----dateset()
--------mini_imagenet()
----img()
--------idea2.png(97KB)
--------idea1.png(95KB)
--------pcl_sinkhorn_framework.png(268KB)
--------idea3.png(100KB)
----simple_sinkhron.py(1KB)
----idea1()
--------main_pcl.py(23KB)
--------pcl()
--------LICENSE(1KB)
--------eval_linear_cls.py(20KB)
----idea3()
--------main_pcl.py(29KB)
--------pcl()
--------LICENSE(1KB)
--------eval_linear_cls.py(20KB)
----README.md(2KB)
----idea2()
--------main_pcl.py(23KB)
--------pcl()
--------LICENSE(1KB)
--------eval_linear_cls.py(20KB)