Image-Cap
所属分类:模式识别(视觉/语音等)
开发工具:Python
文件大小:692KB
下载次数:10
上传日期:2018-04-18 17:02:12
上 传 者:
允耳在心
说明: 通过机器式学习算法完成对图像的识别,并生成相应的文字序列
(Through the machine learning algorithm, the image recognition is completed and the corresponding text sequence is generated.)
文件列表:
Image-Cap (0, 2018-03-27)
Image-Cap\.idea (0, 2018-04-11)
Image-Cap\.idea\Image-Cap.iml (455, 2018-03-13)
Image-Cap\.idea\misc.xml (205, 2018-03-13)
Image-Cap\.idea\modules.xml (270, 2018-03-13)
Image-Cap\.idea\workspace.xml (16592, 2018-04-11)
Image-Cap\caption.py (3965, 2018-03-13)
Image-Cap\const.py (129, 2018-03-13)
Image-Cap\data_loader.py (2599, 2018-03-13)
Image-Cap\generate.py (3311, 2018-03-13)
Image-Cap\model.py (6711, 2018-03-13)
Image-Cap\optim.py (1042, 2018-03-13)
Image-Cap\result (0, 2018-03-27)
Image-Cap\result\.DS_Store (6148, 2018-03-13)
Image-Cap\result\000000046378.jpg (133606, 2018-03-13)
Image-Cap\result\000000063740.jpg (109606, 2018-03-13)
Image-Cap\result\000000221754.jpg (154519, 2018-03-13)
Image-Cap\result\000000228214.jpg (170628, 2018-03-13)
Image-Cap\result\000000249025.jpg (126135, 2018-03-13)
Image-Cap\rouge.py (2892, 2018-03-13)
Image-Cap\train.py (10975, 2018-03-13)
## Introduction
Image Captioning.
Module implemention from "[Actor-Critic Sequence Training for Image Captioning](https://arxiv.org/abs/1706.09601)".
## Requirement
* python 3.5
* pytorch 0.3.1
* numpy 1.13.1
* PIL
## Dataset - COCO
Download the coco datas from [here](http://cocodataset.org/#download). We use train2017 and val2017
## Model training
* Step 1 - Download dataset, move train2017/ val2017/ captions_train2017.json and captions_val2017.json to {MODEL_HOME}/data/
* Step 2 - Run caption.py
```
$ python3 caption.py
```
* Step 3 - Train model
```
$ python3 train.py
```
## Result
My Captioning: Vase with a flower in it sitting on a table.
***
My Captioning: Cat is laying on a couch with a stuffed animal.
***
My Captioning: City street filled with lots of tall buildings.
***
My Captioning: Desk with a computer and a laptop on it.
***
My Captioning: Man wearing a black jacket and a white shirt.
近期下载者:
相关文件:
收藏者: