Tensorflow2.x代码实现计算Top-k Accuracy

图像分类或是识别任务中,一般要求计算top-1,top-2,tor-5等准确率,下面是用Tensorflow2实现这一功能的基本代码,可以根据要求改代码分别计算:

def accuracy(output,target,topk(1,)):
    maxk=max(topk)
    batch_size=target.shape[0]
    
    pred=tf.math.top_k(output,maxk).indices
    pred=tf.transpose(pred,perm=[1,0])
    target_=tf.broadcast_to(target,pred.shape)
    correct=tf.equal(target_,pred)

    res=[]
    for k in topk:
        correct_k=tf.cast(tf.reshape(correct[:k],[-1]),dtype=tf.float32)
        correct_k=tf.reduce_sum(correct_k)
        acc=float(correct_k/batch_size)
        res.append(acc)
    return res

 

上一篇:TensorFlow2 入门指南 | 09 损失函数、梯度下降、优化器选择


下一篇:【深度学习】猫狗识别TensorFlow2实验报告