Github代码复现——SimCLR跑自己的数据集(TensorFlow2)

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

代码链接:添加链接描述

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

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

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

[En]

Let’s take a remote sensing image classification data set as an example.

整体代码结构:

Github代码复现——SimCLR跑自己的数据集(TensorFlow2)

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

Github代码复现——SimCLR跑自己的数据集(TensorFlow2)
Github代码复现——SimCLR跑自己的数据集(TensorFlow2)
新建一个dataset.py文件用于读取数据:
Github代码复现——SimCLR跑自己的数据集(TensorFlow2)

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

Github代码复现——SimCLR跑自己的数据集(TensorFlow2)
在run.py里面添加三个参数,分别为训练样本数,验证样本数,类别总数
Github代码复现——SimCLR跑自己的数据集(TensorFlow2)
在run.py里面对照下面代码进行修改
Github代码复现——SimCLR跑自己的数据集(TensorFlow2)
Github代码复现——SimCLR跑自己的数据集(TensorFlow2)
以上源码
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/497398/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球