History

liaoxingyu 7e83d3175f update README.md Summary: add information about fastreid V1.0		2021-01-18 11:44:55 +08:00
..
ReadMe.md	update pytorchtocaffe tool	2020-06-09 14:45:27 +08:00
__init__.py	add pytorch to caffe converting	2020-06-02 18:11:02 +08:00
caffe.proto	update pytorchtocaffe tool	2020-06-09 14:45:27 +08:00
caffe_lmdb.py	add pytorch to caffe converting	2020-06-02 18:11:02 +08:00
caffe_net.py	update pytorchtocaffe tool	2020-06-09 14:45:27 +08:00
caffe_pb2.py	update pytorchtocaffe tool	2020-06-09 14:45:27 +08:00
layer_param.py	update README.md	2021-01-18 11:44:55 +08:00
net.py	update README.md	2021-01-18 11:44:55 +08:00

ReadMe.md

The Caffe in nn_tools Provides some convenient API

If there are some problem in parse your prototxt or caffemodel, Please replace the caffe.proto with your own version and compile it with command protoc --python_out ./ caffe.proto

caffe_net.py

Using from nn_tools.Caffe import caffe_net to import this model

Prototxt

net=caffe_net.Prototxt(file_name) to open a prototxt file
net.init_caffemodel(caffe_cmd_path='caffe') to generate a caffemodel file in the current work directory
if your caffe cmd not in the $PATH, specify your caffe cmd path by the caffe_cmd_path kwargs.

Caffemodel

net=caffe_net.Caffemodel(file_name) to open a caffemodel
net.save_prototxt(path) to save the caffemodel to a prototxt file (not containing the weight data)
net.get_layer_data(layer_name) return the numpy ndarray data of the layer
net.set_layer_date(layer_name, datas) specify the data of one layer in the caffemodel .datas is normally a list of numpy ndarray [weights,bias]
net.save(path) save the changed caffemodel

Functions for both Prototxt and Caffemodel

net.add_layer(layer_params,before='',after='') add a new layer with Layer_Param object
net.remove_layer_by_name(layer_name)
net.get_layer_by_name(layer_name) or net.layer(layer_name) get the raw Layer object defined in caffe_pb2