文件名称：train_cnn-rnn:cnn + rnn
文件大小：3.08MB
文件格式：ZIP
更新时间：2024-05-20 02:33:34
Python
自己搭建的一个训练框架，包含模型有：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 另外此代码加
【文件预览】：
train_cnn-rnn-master
----pretrain()
--------README.md(40B)
----sample_train()
--------1female()
--------0male()
--------3other()
--------2many()
----config.py(3KB)
----data_aug()
--------data_aug.py(2KB)
--------__pycache__()
----main.py(3KB)
----net()
--------inception_resnet_v2()
--------cifarnet()
--------vgg()
--------z_build_net()
--------resnet_v2()
--------inception_v4()
--------alexnet()
----README.md(24KB)
----load_image()
--------load_image.py(6KB)
--------__pycache__()
----z_ckpt_pb()
--------inception_v4.py(15KB)
--------inception_utils.py(2KB)
--------inception_preprocessing.py(12KB)
--------test.py(2KB)
--------ckpt_pb.py(6KB)
--------img_preprocessing.py(4KB)
----train_net()
--------train.py(9KB)
--------__pycache__()