
颜色分类LeetCode-SimCLRv1-Keras-TensorFlow:基于TensorFlow和Keras的SimCLR实现...
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目为基于TensorFlow与Keras框架下的SimCLR v1实现,专注于利用对比学习进行无监督特征表示的学习,并应用于颜色图像分类任务中。
颜色分类的LeetCode模拟CLRSimCLRv1在TensorFlow-Keras中的实现能够通过视觉表示对比学习(SimCLR)框架提升基础模型的特征表现质量。提供的代码可以将该框架应用于任何Keras模型,只需进行少量修改即可。
给定的实现允许使用5%的数据,在训练线性分类器时提高top-1精度约17%。此外,在应用了SimCLR框架后,t-SNE图显示根据类别对特征进行了清晰的聚类。
可以通过以下步骤重现此结果:
如何使用:
创建SimCLR对象如下:`SimCLR=SimCLR(base_model,input_shape,batch_size,feat_dim,feat_dims_ph,num_of_unfrozen_layers,save_path)`
训练过程可以调用`SimCLR.train`方法,通过传递相应的训练和验证数据来实现。
全部评论 (0)
还没有任何评论哟~


