daquexian / dnntools

[Deprecated] Convert caffemodel to DNNLibrary's daq format

dnntools

dnntools is deprecated and DNNLibrary has switched to ONNX model. Please use onnx2daq in DNNLibrary.

Convert caffemodel to DNNLibrary's format(I named this format "daq" for my ID "daquexian").

Usage

Install this package by pip (It only support Python3.5+ for I used type annotation, and you need install pycaffe first.) :

pip3 install dnntools

Use it in python like this:

import dnntools.caffe_converter

dnntools.caffe_converter.convert('models/squeezenet/deploy.prototxt',
                                 'models/squeezenet/squeezenet_v1.1.caffemodel',
                                 'models/squeezenet/squeezenet.daq')

Not all layers and properties are supported. But I will work on it. What's more I'm working on onnx support.

Any PRs are welcome :)

Have a problem?

Some users met segment fault when converting models. One of the possible reasons is incompatible protobuf version. Please refer to this issue.