Providing a tool for some fashion neural network frameworks.
The nn_tools is released under the MIT License (refer to the LICENSE file for details).
- Converting a model between different frameworks.
- Constructing a model using this tools quickly.
- Analysing a model, get the operations number(ops) in every layers.
- Python2.7
- Each functions in this tools requires corresponding neural network python package (tensorflow pytorch and so on).
Before you analyse your network, Netscope is recommended to visiualize your network.
Command:python caffe_analyser.py [-h] prototxt outdir shape
- The prototxt is the path of the prototxt file.
- The outdir is path to save the csv file.
- The shape is the input shape of the network(split by comma
,
), image shape should be: batch_size, image_height, image_width, channel.
For example python caffe_analyser.py resnet_18_deploy.prototxt analys_result.csv 1,224,224,3
doc comming soon
doc comming soon