Search icon CANCEL
Arrow left icon
Explore Products
Best Sellers
New Releases
Books
Videos
Audiobooks
Learning Hub
Conferences
Free Learning
Arrow right icon
Arrow up icon
GO TO TOP
Caffe2 Quick Start Guide

You're reading from   Caffe2 Quick Start Guide Modular and scalable deep learning made easy

Arrow left icon
Product type Paperback
Published in May 2019
Publisher Packt
ISBN-13 9781789137750
Length 136 pages
Edition 1st Edition
Languages
Tools
Arrow right icon
Author (1):
Arrow left icon
Ashwin Nanjappa Ashwin Nanjappa
Author Profile Icon Ashwin Nanjappa
Ashwin Nanjappa
Arrow right icon
View More author details
Toc

Converting a Caffe model to Caffe2

To be able to use a Caffe model in Caffe2, we need to convert it from its Caffe formats to Caffe2 file formats. Caffe2 provides a script named python/caffe_translator.py that can be used for this purpose.

For example, we can convert our AlexNet files from Caffe to Caffe2 by invoking the script as follows:

$ python python/caffe_translator.py path_to_caffe/models/bvlc_alexnet/deploy.prototxt path_to_caffe/models/bvlc_alexnet/bvlc_alexnet.caffemodel --init_net init_net.pb --predict_net predict_net.pb

Running this script generates three files, predict_net.pb, predict_net.pbtxt, and init_net.pb, for AlexNet:

Figure 4.3: AlexNet network structure in Caffe2

Figure 4.3 shows the AlexNet network structure in Caffe2 after it was converted from the Caffe model. This graph visualization was generated using the Caffe2 net_drawer.py tool that utilizes GraphViz...

lock icon The rest of the chapter is locked
Register for a free Packt account to unlock a world of extra content!
A free Packt account unlocks extra newsletters, articles, discounted offers, and much more. Start advancing your knowledge today.
Unlock this book and the full library FREE for 7 days
Get unlimited access to 7000+ expert-authored eBooks and videos courses covering every tech area you can think of
Renews at €18.99/month. Cancel anytime