tensorflow center loss代码注释
来源:互联网 发布:数据库系统的构成 编辑:程序博客网 时间:2024/04/30 15:39
def center_loss(features, label, alfa, nrof_classes): """Center loss based on the paper "A Discriminative Feature Learning Approach for Deep Face Recognition" (http://ydwen.github.io/papers/WenECCV16.pdf) """ #获取特征向量长度 nrof_features = features.get_shape()[1] #生成可以共享的变量centers,由于center loss在计算图中只存在于一个节点处,因此这个变量只使用一次 #不需要所谓的variable_scope,就可以实现每次共享? centers = tf.get_variable('centers', [nrof_classes, nrof_features], dtype=tf.float32, initializer=tf.constant_initializer(0), trainable=False) label = tf.reshape(label, [-1]) #取出对应label下对应的center值,注意label里面的值可能会重复,因为一个标签下有可能会出现多个人 centers_batch = tf.gather(centers, label) #求特征点到中心的距离并乘以一定的系数,alfa是center的更新速度,越大代表更新的越慢 diff = (1 - alfa) * (centers_batch - features) #更新center,输出是将对应于label的centers减去对应的diff,如果同一个标签出现多次,那么就减去多次 centers = tf.scatter_sub(centers, label, diff) #求center loss,这里是将l2_loss里面的值进行平方相加,再除以2,并没有进行开方 loss = tf.nn.l2_loss(features - centers_batch) return loss, centers
1 1
- tensorflow center loss代码注释
- TensorFlow实现center loss
- TensorFlow实现center loss
- center loss代码注释(caffe新添加层)
- Center Loss层
- tensorflow示例代码注释1
- tensorflow示例代码注释2
- tensorflow示例代码注释3
- tensorflow示例代码注释4
- tensorflow示例代码注释5
- tensorflow:mnis入门代码注释
- tensorflow:深入mnist代码注释
- caffe 训练center loss 太大
- 对Center loss的理解
- tensorflow计算Loss
- tensorflow:fully_connected_feed.py代码详细中文注释
- 损失函数改进之Center Loss
- tensorflow从变量-loss-optimizer
- bzoj2434 [Noi2011]阿狸的打字机
- react-navigation之动态修改title的内容
- 基于DirectUI技术开发的发卡系统
- java的TreeMap在ip转区域的使用
- 弹出框页面居中显示的两种方法
- tensorflow center loss代码注释
- 史上最全的CSS hack方式一览
- 初识JHipster
- 接上--根据不同请求路径处理不同响应
- opencv图像阈值设置
- 马景涛手写长文宣布离婚 表示放飞妻女送上祝福十分咆哮
- Akka TypedActor
- Java学习之switch
- Linux内核与根文件系统的关系