weixin_43864917的博客

Github代码复现——自监督学习SimCLR跑自己的数据集（TensorFlow2）代码链接：添加链接描述框架用的是tensorflow2，这里就不详细说明SimCLR的原理和环境配置了。原版代码里面用的是tensorflow-dataset下载并读取数据集，如果你的数据集tensorflow-dataset里面正好有，那就直接按照github上的说明进行训练。这里我们讲如何训练自己的数据。我们以一个遥感图像分类数据集为例整体代码结构：命名一个data文件夹，用于存放训练数据和验证数据。