基本信息

案例ID:132610

技术顾问:Z_明晃晃 - 1年经验 - 中原银行

联系沟通

微信扫码,建群沟通

项目名称:images story

所属行业:人工智能 - 其他

->查看更多案例

案例介绍

Pytorch image_caption生成字幕功能模块

数据来自于AI Challenger图像描述,需要安装pytorh和requirements.txt环境依赖,可以用data_preprocess.py进行新模型的训练。
预处理python data_preprocess.py process
--annotation-file=/data/annotation.json
--max-words=5000
​提取图片特征python feature_extract.py
​训练 python main.py train
​​​​--img_path='path to train_image'
​​​​--img_features_path='../results.pth'
​成果

GPT2_chinese故事生成功能模块

中文的GPT2训练代码,可以写诗,新闻,小说,或是训练通用语言模型。使用Bert的tokenizer处理中文字符。
生成样例

相似案例推荐

其他人才的相似案例推荐

发布任务

企业点击发布任务,工程师会在任务下报名,招聘专员也会在1小时内与您联系,1小时内精准确定人才

微信接收人才推送

关注猿急送微信平台,接收实时人才推送

接收人才推送
联系需求方端客服
联系需求方端客服