/AutomaticAnnotation

Primary LanguageCudaApache License 2.0Apache-2.0

项目简述

识别广告图片中的产品和文字,自动生成结构化数据集,详细信息见Our Website

环境

  • Python2.7
  • cython
  • Caffe

环境配置参考:FCISCTPN

模型下载:

产品识别模型(大约233M)下载:OneDrive或者BaiduYun(密码为tmd4),把模型放入文件夹 model中并重命名为fcis_coco-0000.params

字体识别模型 (大约78M) 下载:Google Drive, 把模型放入文件夹 models中并重命名为 ctpn_trained_model.caffemodel

运行方式

python ./segmentation/infoExtraction.py

Enjoy It!