下载CUB200对应的caption标签

最新推荐文章于 2024-11-15 10:17:44 发布

卡卡卡卡NI

最新推荐文章于 2024-11-15 10:17:44 发布

阅读量255

点赞数

文章标签： python 开发语言

本文链接：https://blog.csdn.net/skn000/article/details/129867800

版权

文章介绍了如何使用PyTorch加载CUB200数据集中metadata.pth文件，将txt文件中的词ID转换为实际单词，涉及到数据预处理和文本处理技术。

摘要由CSDN通过智能技术生成

kaggle这个网页介绍了CUB200数据的主要内容CUB 200 2011 | Kaggle

在这个地方对应的是caption标签的txt文件，把这个txt文本下载下来

下载下来的txt里面是word的id，然后根据网页的介绍，这里有一个metadata.pth文件，用来把id转成word,把这个metadata.pth也下载下来。

下面是把id转成word的代码过程。

import torch
metadata=torch.load("D:\研究生生活实录\数据资料\CUB_200_2011\metadata.pth\metadata.pth")

with open("D:\研究生生活实录\数据资料\CUB_200_2011\cub_200_2011_all_caps.txt")as f:
    rawtext=f.readlines()
 
newtext=[]

for sentence in rawtext:
    
    tempnewsentence=[]
    sentence=sentence.replace("\n",'')
    sentence=list(sentence.split(' '))
    for word_id in sentence:
        if word_id=='0':
            break
        word=metadata['word_id_to_word'][int(word_id)]
        tempnewsentence.append(word)
    newsentence=' '.join(w for w in tempnewsentence )
    newtext.append(newsentence)

转之前的样子