eval_coco.py 文件源码

def eval_coco(annFile, resFile):

    coco = COCO(annFile)
    cocoRes = coco.loadRes(resFile)
    cocoEval = COCOEvalCap(coco, cocoRes)
    cocoEval.evaluate()

    Bleu_4 = cocoEval.eval['Bleu_4']
    METEOR = cocoEval.eval['METEOR']
    ROUGE_L = cocoEval.eval['ROUGE_L']
    CIDEr = cocoEval.eval['CIDEr']
    total = Bleu_4 + METEOR + ROUGE_L + CIDEr
    score = {'Bleu_4': Bleu_4, 'METEOR': METEOR, 'ROUGE_L': ROUGE_L, 'CIDEr': CIDEr, 'total': total}

    return score