文件名称:nn_tools:神经网络工具
文件大小:104KB
文件格式:ZIP
更新时间:2024-05-02 15:06:04
converter caffe deep-learning pytorch analyzer
神经网络工具:转换器,构造函数和分析器 提供一些时尚神经网络框架的工具。 nn_tools是根据MIT许可证发布的(有关详细信息,请参阅LICENSE文件)。 特征 在不同框架之间转换模型。 一些便捷的工具可以快速操作caffemodel和prototxt(例如获取或设置图层的权重),请参见 。 分析模型,获取每层中的工序数。 要求 Python2.7或Python3.x 此工具中的每个功能都需要相应的神经网络python软件包(tensorflow pytorch等)。 转炉 Pytorch到Caffe pytorch_to_caffe的新版本支持pytorch的最新版本(从0.2.0到1.0)。 注意:由于实现上的差异,传输输出与原始模型会有所不同。 支持的图层类型:conv2d,transpose_conv2d,线性,max_pool2d,avg_pool2d,辍学,re
【文件预览】:
nn_tools-master
----.gitignore(79B)
----Pytorch()
--------augmentations.py(19KB)
--------__init__.py(39B)
--------eval.py(2KB)
--------utils.py(3KB)
--------train.py(3KB)
----analysis()
--------__init__.py(81B)
--------blob.py(2KB)
--------roi.py(1010B)
--------layers.py(12KB)
--------utils.py(3KB)
--------CaffeA.py(5KB)
--------README.md(95B)
--------PytorchA.py(3KB)
--------MxnetA.py(5KB)
----__init__.py(0B)
----caffe_analyser.py(1KB)
----mxnet_analyser.py(2KB)
----LICENSE(1KB)
----keras_to_caffe.py(989B)
----pytorch_analyser.py(2KB)
----Tensorflow()
--------__init__.py(0B)
--------constructor.py(4KB)
--------graph.py(180B)
----Caffe()
--------__init__.py(0B)
--------ReadMe.md(1KB)
--------caffe.proto(77KB)
--------net.py(97B)
--------caffe_lmdb.py(984B)
--------caffe_net.py(5KB)
--------layer_param.py(4KB)
--------caffe_pb2.py(350KB)
----README.md(5KB)
----example()
--------inceptionv3_pytorch_to_caffe.py(497B)
--------vgg19_pytorch_to_caffe.py(448B)
--------testify_pytorch_to_caffe_example.py(5KB)
--------resnet_pytorch_to_caffe.py(454B)
--------densenet_pytorch_to_caffe.py(484B)
--------alexnet_pytorch_to_caffe.py(458B)
--------resnet_pytorch_analysis_example.py(406B)
----Datasets()
--------base_datasets.py(956B)
--------lmdb_data.proto(239B)
--------imdb_wiki.py(5KB)
--------__init__.py(0B)
--------adience.py(3KB)
--------mnist.py(4KB)
--------README.md(403B)
--------lmdb_datasets.py(3KB)
--------cifar10.py(73B)
--------imagenet.py(4KB)
--------wider.py(4KB)
--------lmdb_data_pb2.py(3KB)
----pytorch_to_caffe.py(34KB)
----funcs.py(2KB)