-
train_cnn-rnn:cnn + rnn下载
资源介绍
自己搭建的一个训练框架,包含模型有:cnn+rnn: vgg(vgg16,vgg19)+rnn(LSTM, GRU), resnet(resnet_v2_50,resnet_v2_101,resnet_v2_152)+rnn(LSTM, GRU), inception_v4+rnn(LSTM, GRU), inception_resnet_v2+rnn(LSTM, GRU)等。
此框架主要针对分类任务, 后面会陆续搭建多任务多标签、检测等框架,欢迎关注。
使用说明:
搭建时使用的环境为:Python3.5, tensorflow1.4
变量设置参考config.py。
详细说明参见config.py。
( mkdir pretrain/inception_v4, 下载与训练模型, cp到pretrain/inception_v4/ )
运行代码: python main.py
另外此代码加
- 上一篇: ctc-process:CTC流程模型
- 下一篇: SLIC超像素分割 matlab版本的代码