Add ability to import Caffe networks

Question

Add ability to import Caffe networks

alejandro-isaza opened this issue 9 years ago · 13 comments

I don't think we should add all of Protobuf but maybe have a script that converts a caffe prototxt into simpler format. Then have that simpler format be the standard BrainCore network definition format.

Answer 1 · 2015-10-24T20:48:18.000Z

@Aleph7 Are you thinking JSON? Or something even lighter? Take a look at msgpack.org

Answer 2 · 2015-10-24T22:22:26.000Z

I was thinking YAML

-
    name: ip1
    type: Inner Product
-
    name: relu1
    type: ReLU
    sources: [ip1]

Answer 3 · 2015-10-25T19:01:37.000Z

Ooo very nice.

Answer 4 · 2016-04-09T21:54:07.000Z

Any news on that? Will love to use Caffe models with BrainCore. 👍

Answer 5 · 2016-04-10T17:52:24.000Z

@hartator this is not a priority for us right now, hence the 'help wanted' tag.

Answer 6 · 2016-05-12T20:30:29.000Z

you maybe able to cherry pick from this code
https://github.com/naru-hiyoko/simple-cnn-swift

https://github.com/noradaiko/caffe-ios-sample

Answer 7 · 2016-05-26T08:03:05.000Z

https://github.com/DeepLearningKit/caffemodel2json could maybe help

Answer 8 · 2016-08-10T17:15:20.000Z

@Aleph7 since BrainCore primary target is mobile, wouldn't it be more reasonable to drop json/yaml and go with binary protobuf, to avoid processing overhead?

Answer 9 · 2016-08-17T02:15:19.000Z

@skeeet yes, that would be good but I don't want to force people to get all the protobuf dependencies just to use BrainCore. Maybe that can be a separate project.

Answer 10 · 2016-09-27T17:47:23.000Z

Maybe we should just read protobuf now that https://github.com/apple/swift-protobuf-plugin is out!

Answer 11 · 2017-01-30T20:14:36.000Z

Hi, I'm wondering if this is still an ongoing update to BrainCore still? I'm interested in possibly helping to contribute but I'm not sure where to get started.

Thanks!

Answer 12 · 2017-01-30T20:24:56.000Z

If you want to contribute reading protobuf models I would suggest starting with caffe's protobuf definition. Try to to generate the Swift code for that with with swift-protobuf on a project by itself. Then try generating BrainCore layers from that. If you get that working on a separate repo I'd be happy to integrate it to BrainCore.

Answer 13 · 2017-02-15T00:45:23.000Z

Great thanks, I'll take a look at it.