Skip to content

codeaudit/caffe-tensorflow

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Caffe to TensorFlow

This is a proof-of-concept that shows how to use pretrained Caffe models in TensorFlow.

This demo imports the VGG-16 network network.

Verified on the ILSVRC2012 validation set, on which it achieves a top 5 accuracy of 86.64%.

A minor quirk: you need to run kaffe.py to extract the parameters first, since it appears that Caffe and TensorFlow don't play nice (CUDA conflicts). This persists even with set_mode_cpu. However, for future work, there's really no need to depend on Caffe here: just the protocol buffers should suffice.

About

Proof of concept for loading Caffe models in TensorFlow

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%