登录 注册
当前位置:主页 > 资源下载 > 7 > train_cnn-rnn:cnn + rnn下载

train_cnn-rnn:cnn + rnn下载

  • 更新:2024-09-14 22:25:50
  • 大小:3.08MB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

自己搭建的一个训练框架,包含模型有: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 另外此代码加