README.md 876 字节
Newer Older
L
liuyang11 已提交
1 2
### Caffe2Fluid
This tool is used to convert a Caffe model to Fluid model
3

L
liuyang11 已提交
4 5 6 7
### Howto
1, Prepare caffepb.py in ./proto, two options provided
    1) generate it from caffe.proto using protoc
        bash ./proto/compile.sh
8

L
liuyang11 已提交
9 10
    2) download one from github directly
        cd proto/ && wget https://github.com/ethereon/caffe-tensorflow/blob/master/kaffe/caffe/caffepb.py
11

W
wanglong03 已提交
12
2, Convert the caffe model using 'convert.py' which will generate a python script and a weight(in .npy) file
13

L
liuyang11 已提交
14 15
3, Use the converted model to predict
    see more detail info in 'tests/lenet/README.md'
16 17


L
liuyang11 已提交
18 19
### Supported models
- Lenet on mnist dataset
20

L
liuyang11 已提交
21 22
- ResNets:(ResNet-50, ResNet-101, ResNet-152)
    model addrs:(https://onedrive.live.com/?authkey=%21AAFW2-FVoxeVRck&id=4006CBB8476FF777%2117887&cid=4006CBB8476FF777)
23

L
liuyang11 已提交
24 25
### Notes
Some of this code come from here: https://github.com/ethereon/caffe-tensorflow