Published by Packt
This is the code repository for [Caffe2 Quick Start Guide](Packt UTM URL of the Book), published by Packt.
Modular and scalable deep learning made easy
Caffe2 is a popular deep learning library used for fast and scalable training and inference of deep learning models on various platforms. This book introduces you to the Caffe2 framework and shows how you can leverage its power to build, train, and deploy efficient neural network models at scale.
This book covers the following exciting features:
- Build and install Caffe2
- Compose neural networks
- Train neural network on CPU or GPU
- Import a neural network from Caffe
- Import deep learning models from other frameworks
If you feel this book is for you, get your copy today!
All of the code is organized into folders. For example, Chapter02.
The code will look like the following:
# Cast grayscale pixel values to float
# Scale pixel values to [0, 1]
input_images = model.Cast(input_images_uint8, "input_images",
to=core.DataType.FLOAT)
input_images = model.Scale(input_images, input_images, scale=float(1./256))
Following is what you need for this book:
Data scientists and machine learning engineers who wish to create fast and scalable deep learning models in Caffe2 will find this book to be very useful. Some understanding of the basic machine learning concepts and prior exposure to programming languages like C++ and Python will be useful.
With the following software and hardware list you can run all code files present in the book (Chapter 01-07).
| Chapter | Software required | OS required |
|---|---|---|
| 1-7 | NVIDIA GPU driver | Windows, Mac OS X, and Linux (Any) |
| CUDA | ||
| Intel Math Kernel Library |
Click on the following link to see the Code in Action:
Ashwin Nanjappa is a senior architect at NVIDIA, working in the TensorRT team on improving deep learning inference on GPU accelerators. He has a PhD from the National University of Singapore in developing GPU algorithms for the fundamental computational geometry problem of 3D Delaunay triangulation. As a post-doctoral research fellow at the BioInformatics Institute (Singapore), he developed GPU-accelerated machine learning algorithms for pose estimation using depth cameras. As an algorithms research engineer at Visenze (Singapore), he implemented computer vision algorithm pipelines in C++, developed a training framework built upon Caffe in Python, and trained deep learning models for some of the world's most popular online shopping portals.
Click here if you have any feedback or suggestions.
If you have already purchased a print or Kindle version of this book, you can get a DRM-free PDF version at no cost.
Simply click on the link to claim your free PDF.