blob: 2a5e350e4afb8975563b6f1febec34a2ff253df1 [file] [log] [blame] [view]
# Image Classification Example Using the C Predict API
This is a simple predictor which shows how to use the MXNet C Predict API for image classification with a pre-trained ImageNet model in a single thread and multiple threads.
## Prerequisites
* OpenCV for image processing: `USE_OPENCV` is set to true by default when [building from source](https://mxnet.incubator.apache.org/install/build_from_source.html)
## How to Use this Example
### Download the Model Artifacts
1. You will need the model artifacts for the Inception ImageNet model. You can download these from http://data.mxnet.io/mxnet/models/imagenet/inception-bn/
2. Place them into a `model/Inception/` subfolder, or if not, you will need to edit the source file and update the paths in the Build step.
* [model/Inception/Inception-BN-symbol.json](http://data.mxnet.io/mxnet/models/imagenet/inception-bn/Inception-BN-symbol.json)
* [model/Inception/Inception-BN-0126.params](http://data.mxnet.io/mxnet/models/imagenet/inception-bn/Inception-BN-0126.params)
* [model/Inception/synset.txt](http://data.mxnet.io/mxnet/models/imagenet/synset.txt)
### Build
1. If using a different location for the model artifacts, edit `image-classification-predict.cc` file, and change the following lines to your artifacts' paths:
```c
// Models path for your model, you have to modify it
std::string json_file = "model/Inception/Inception-BN-symbol.json";
std::string param_file = "model/Inception/Inception-BN-0126.params";
std::string synset_file = "model/Inception/synset.txt";
std::string nd_file = "model/Inception/mean_224.nd";
```
2. You may also want to change the image size and channels:
```c
// Image size and channels
int width = 224;
int height = 224;
int channels = 3;
```
3. Simply just use our Makefile to build:
```bash
make
```
### Run
Run the example by passing it an image that you want to classify. If you don't have one handy, run the following to get one:
```bash
wget https://upload.wikimedia.org/wikipedia/commons/thumb/f/f4/Honeycrisp.jpg/1920px-Honeycrisp.jpg
```
Then run the `image-classification-predict` program, passing the image as the first argument and the number of threads as the second parameter.
```bash
./image-classification-predict 1920px-Honeycrisp.jpg 1
```
## Tips
* If you don't run it in the MXNet root path, you may need to copy the `lib` folder here.
## Author
* **Xiao Liu**
* E-mail: liuxiao@foxmail.com
* Homepage: [www.liuxiao.org](http://www.liuxiao.org/)
## Thanks
* pertusa (for Makefile and image reading check)
* caprice-j (for reading function)
* sofiawu (for sample model)
* piiswrong and tqchen (for useful coding suggestions)