word_count_mean of COCO Caption

[moiSusu]

How is the "word_count_mean" in the COCO Caption results computed? I can't find it in the scripts.

[kennymckormick]

Hi, @moiSusu , word_count_mean is counted based on the prediction files of COCO_VAL, and the script is not included in this repo. Below is the function we used for word counting, in case you need:

def word_count(inText):
    inText = str(inText)
    outText = inText
    punct = [';', r'/', '[', ']', '"', '{', '}', '(', ')', '=', '+', '\\', '_', '-', '>', '<', '@', '`', ',', '?', '!']
    commaStrip = re.compile('(\d)(,)(\d)')  # noqa: W605
    periodStrip = re.compile('(?!<=\d)(\.)(?!\d)')  # noqa: W605
    for p in punct:
        if (p + ' ' in inText or ' ' + p in inText) or (re.search(
                commaStrip, inText) is not None):
            outText = outText.replace(p, '')
        else:
            outText = outText.replace(p, ' ')
    outText = periodStrip.sub('', outText, re.UNICODE)
    return len(outText.split())