对image caption任务做evaluation(计算Bleu1,Bleu2,Bleu3,Bleu4,METEOR,ROUGE_L,CIDEr)

用到一两个image caption任务的模型,模型中没有给出性能评估的方法,于是在github上找到了一个能够评估image caption的模型,该模型能够帮助计算常见的评价指标的结果,包括Bleu1Bleu2Bleu3Bleu4METEORROUGE_LCIDErSkipThoughtsCosineSimilarityEmbeddingAverageCosineSimilarityVectorExtremaCosineSimilarityGreedyMatchingScore
github地址:点击链接

依照github上的要求一步步安装后,在nlg-eval-master根目录上导入result.txt(生成的caption),caption1.txt(caption的标准结果1),caption2.txt(caption的标准结果2),caption3.txt(caption的标准结果3),caption4.txt(caption的标准结果4),caption5.txt(caption的标准结果5),并创建文件evaluation.py

import nlgeval
res=nlgeval.compute_metrics(hypothesis='./result.txt',
                   references=['./caption1.txt','./caption2.txt','./caption3.txt','./caption4.txt','./caption5.txt'])
print(res)

就会得到各评价指标的结果。

  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值