Github代码复现——SimCLR跑自己的数据集（TensorFlow2）

Github代码复现——自监督学习SimCLR跑自己的数据集（TensorFlow2）

代码链接：添加链接描述

框架用的是tensorflow2，这里就不详细说明SimCLR的原理和环境配置了。

原版代码里面用的是tensorflow-dataset下载并读取数据集，如果你的数据集tensorflow-dataset里面正好有，那就直接按照github上的说明进行训练。这里我们讲如何训练自己的数据。

我们以一个遥感图像分类数据集为例

整体代码结构：

命名一个data文件夹，用于存放训练数据和验证数据。


新建一个dataset.py文件用于读取数据：

按照下面的代码修改原来data.py

在run.py里面添加三个参数，分别为训练样本数，验证样本数，类别总数

在run.py里面对照下面代码进行修改


以上源码：
https://download.csdn.net/download/weixin_43864917/85044850

付费资源根据自身情况下载吧。稍微懂点的，参考我的博客，自己改改肯定是可以跑起来的。