基于NeuralTalk2 图片标注 - web版

web版图片标注示例如有问题，欢迎拍砖，多多指教 😀

简要说明

本案例主要使用nerualtalks基于mscoco数据集训练后的模型,搭建flask网站进行上传图片进行标注,具体配置可参考neuraltalk2进行操作.

环境配置

ubuntu16 + python3 + lua + torch + flask

具体使用

git clone https://github.com/RookieDay/neuraltalk2_web.git
下载对应依赖包/配置环境
根据neuraltalk2 readme 下载对应的model checkpoint（有GPU版和CPU版本）
python app.py --debug True (debug模式默认使用CPU)
浏览器输入 http://0.0.0.0:5000/ 即可

网站效果预览

开发计划

上传单张图片标注；
实现上传多张图片标注；
实现上传视频标注；
添加进度条；

许可

MIT © RookieDay