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
付费资源根据自身情况下载吧。稍微懂点的,参考我的博客,自己改改肯定是可以跑起来的。