文件名称:Image-Captioning:用Pytorch进行图像字幕
文件大小:95KB
文件格式:ZIP
更新时间:2024-05-04 17:38:53
Python
图片字幕 介绍 该项目是论文。 它可能并不完全相似。 使用Pytorch作为代码。 ResNet101用于提取功能。 您可以检查预训练的模型。 使用2017 Val图像[5K / 1GB],注释[241MB]。 请检查make_vocab.py和data_loader.py。 Vocab.pickle是一个pickle文件,其中包含注释中的所有单词。 coco_ids.npy存储要使用的图像ID。 另外,您必须设置路径或其他设置。 执行prerocess_idx函数。 您可以运行源代码并尝试自己的示例。 环境 的Python 3.8.5 火炬1.7.1 CUDA 11.0 如何使用 火车 cd src python train.py 进行测试 cd src python sample.py 结果 时代100 说明文字:一名妇女在手提箱中手持一只泰迪熊 待办事项清单 张量
【文件预览】:
Image-Captioning-master
----src()
--------sample.py(3KB)
--------train.py(5KB)
--------make_vocab.py(2KB)
--------model.py(2KB)
--------data_loader.py(3KB)
----LICENSE(1KB)
----README.md(1KB)
----image()
--------000000435205.jpg(87KB)
----.gitignore(10B)