AM-Softmax的keras实现: 《Additive Margin Softmax for Face Verification》

原答案在对输入进行归一化时有错误,另外m应该是一个固定的超参数不需要训练,已改正论文地址:Additive Margin Softmax for Face Verification 和L-Softmax、A-Softmax比简单很多,在minist上测试通过,两个500节点的全连接层 + 10个节点的分类层,训练两轮验证集的acc达到97+%。运行环境: keras:2.1.5...
摘要由CSDN通过智能技术生成

原答案在对输入进行归一化时有错误,另外m应该是一个固定的超参数不需要训练,已改正


论文地址:Additive Margin Softmax for Face Verification
和L-Softmax、A-Softmax比简单很多,在minist上测试通过,两个500节点的全连接层 + 10个节点的分类层,训练两轮验证集的acc达到97+%。


运行环境:
keras:2.1.5
tensorflow:1.4.0
python:3.6
OS:win7


# -*- coding: utf-8 -*-
from keras import backend as K
from keras.engine.topology import Layer
from keras.layers import Dense, Activation,BatchNormalization
from keras.layers import activations, initializers, regularizers, constraints, Lambda
from keras.engine import InputSpec
import tensorflow as tf
import numpy as np

class AMSoftmax(Layer):
    def __init__(self, units, s, m,
                 kernel_initializer='glorot_uniform',
                 kernel_regularizer=None,
                 kernel_constraint=None,
                 **kwargs
                 ):
        if 'input_shape' 
  • 7
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 51
    评论
评论 51
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值