文件名称:caffe-theano-conversion
文件大小:17.1MB
文件格式:ZIP
更新时间:2024-07-06 02:02:56
Python
caffe_theano_conversion 这是斯坦福大学 CS231N 项目的一部分,作者是 Ankit Kumar,数学专业,2015 届 这是一个存储库,允许您将预训练的 caffe 模型转换为 Lasagne 中的模型,Lasagne 是 Theano 的薄包装器。 您还可以将 caffe 模型的架构转换为 Lasagne 中的等效架构。 您不需要安装 caffe 即可使用此模块。 目前,支持以下 caffe 层: * Convolution * LRN * Pooling * Inner Product * Relu * Softmax 您还可以使用转换的 convert_mean_file 函数加载平均文件。 未来的工作是把它放在网络架构本身的转换中,它可能有一个带有转换参数的数据层。 那可以自动化。 现在,您必须将成本层放在自己身上,并进行反向传播代码。 这对我
【文件预览】:
caffe-theano-conversion-master
----MIT-scenes()
--------test_key.pem(2KB)
--------cc_layers.py(17KB)
--------dataset.py(10KB)
--------dump_features.py(3KB)
--------train_baseline.py(8KB)
--------layers.py(66KB)
--------solver.py(8KB)
--------script.py(221B)
----README.md(4KB)
----.gitignore(680B)
----caffe2theano()
--------models()
--------tests.py(4KB)
--------parsing()
--------__init__.py(114B)
--------old.py(6KB)
--------conversion.py(12KB)
--------todo.txt(550B)
--------data()
--------tests()
--------caffe_layers()