标签:Tensorflow2 target pred Top top topk tf correct 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,target,pred,Top,top,topk,tf,correct,Accuracy 来源: https://blog.51cto.com/u_15242250/2870181
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。