Github代码复现——自监督学习SimCLR跑自己的数据集（TensorFlow2）
代码链接：添加链接描述
框架用的是tensorflow2，这里就不详细说明SimCLR的原理和环境配置了。
原版代码里面用的是tensorflow-dataset下载并读取数据集，如果你的数据集tensorflow-dataset里面正好有，那就直接按照github上的说明进行训练。这里我们讲如何训练自己的数据。
让我们以一个遥感图像分类数据集为例。
[En]
Let’s take a remote sensing image classification data set as an example.
整体代码结构：
命名一个data文件夹，用于存放训练数据和验证数据。
新建一个dataset.py文件用于读取数据：
按照下面的代码修改原来data.py
在run.py里面添加三个参数，分别为训练样本数，验证样本数，类别总数
在run.py里面对照下面代码进行修改
以上源码：
https://download.csdn.net/download/weixin_43864917/85044850
根据您自己的情况下载付费资源。如果你知道一点，参考我的博客，如果你自己改变它，你肯定可以跑。
[En]
Download paid resources according to your own situation. If you know a little bit, refer to my blog, you can definitely run if you change it by yourself.
Original: https://blog.csdn.net/weixin_43864917/article/details/123772261
Author: 拖延癌晚期患者7777
Title: Github代码复现——SimCLR跑自己的数据集（TensorFlow2）
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/497388/
转载文章受原作者版权保护。转载请注明原作者出处！