DeepDetect (http://www.deepdetect.com/) is a machine learning API and server written in C++11. It makes state of the art machine learning easy to work with and integrate into existing applications.
DeepDetect relies on external machine learning libraries through a very generic and flexible API. At the moment it has support for:
- the deep learning library Caffe
- distributed gradient boosting library XGBoost
- the deep learning and other usages library Tensorflow
- clustering with T-SNE
Training | Prediction | Classification | Object Detection | Regression | Autoencoder | |
---|---|---|---|---|---|---|
Caffe | Y | Y | Y | Y | Y | Y |
XGBoost | Y | Y | Y | N | Y | N/A |
Tensorflow | N | Y | Y | N | N | N |
T-SNE | Y | N/A | N/A | N/A | N/A | N/A |
Training | Prediction | |
---|---|---|
Caffe | Y | Y |
XGBoost | Y | N |
Tensorflow | Y | Y |
T-SNE | N | N |
CSV | SVM | Text words | Text characters | Images | |
---|---|---|---|---|---|
Caffe | Y | Y | Y | Y | Y |
XGBoost | Y | Y | Y | N | N |
Tensorflow | N | N | N | N | Y |
T-SNE | Y | N | N | N | Y |
(*) more input support for T-SNE is pending |
DeepDetect implements support for supervised and unsupervised deep learning of images, text and other data, with focus on simplicity and ease of use, test and connection into existing applications. It supports classification, object detection, regression, autoencoders, ...
Please join either the community on Gitter or on IRC Freenode #deepdetect, where we help users get through with installation, API, neural nets and connection to external applications.
The reference platforms with support are Ubuntu 14.04 LTS and Ubuntu 16.04 LTS.
Supported images that come with pre-trained image classification deep (residual) neural nets:
-
docker images for CPU and GPU machines are available at https://hub.docker.com/r/beniz/deepdetect_cpu/ and https://hub.docker.com/r/beniz/deepdetect_gpu/ respectively. See https://github.com/beniz/deepdetect/tree/master/docker/README.md for details on how to use them.
-
For Amazon AMI see official builds documentation at https://deepdetect.com/products/ami/, and direct links to GPU AMI and CPU AMI.
Setup an image classifier API service in a few minutes: http://www.deepdetect.com/tutorials/imagenet-classifier/
List of tutorials, training from text, data and images, setup of prediction services, and export to external software (e.g. ElasticSearch): http://www.deepdetect.com/tutorials/tutorials/
Current features include:
- high-level API for machine learning and deep learning
- Support for Caffe, Tensorflow, XGBoost and T-SNE
- classification, regression, autoencoders, object detection
- JSON communication format
- remote Python client library
- dedicated server with support for asynchronous training calls
- high performances, benefit from multicore CPU and GPU
- connector to handle large collections of images with on-the-fly data augmentation (e.g. rotations, mirroring)
- connector to handle CSV files with preprocessing capabilities
- connector to handle text files, sentences, and character-based models
- connector to handle SVM file format for sparse data
- range of built-in model assessment measures (e.g. F1, multiclass log loss, ...)
- no database dependency and sync, all information and model parameters organized and available from the filesystem
- flexible template output format to simplify connection to external applications
- templates for the most useful neural architectures (e.g. Googlenet, Alexnet, ResNet, convnet, character-based convnet, mlp, logistic regression)
- support for sparse features and computations on both GPU and CPU
- Full documentation is available from http://www.deepdetect.com/overview/introduction/
- API documentation is available from http://www.deepdetect.com/api/
- FAQ is available from http://www.deepdetect.com/overview/faq/
- Python client:
- REST client: https://github.com/beniz/deepdetect/tree/master/clients/python
- 'a la scikit' bindings: https://github.com/ArdalanM/pyDD
- Java client: https://github.com/kfadhel/deepdetect-api-java
- Early C# client: https://github.com/beniz/deepdetect/pull/98
- C++, gcc >= 4.8 or clang with support for C++11 (there are issues with Clang + Boost)
- eigen for all matrix operations;
- glog for logging events and debug;
- gflags for command line parsing;
- OpenCV >= 2.4
- cppnetlib
- Boost
- curl
- curlpp
- utfcpp
- gtest for unit testing (optional);
- CUDA 8 or 7.5 is recommended for GPU mode.
- BLAS via ATLAS, MKL, or OpenBLAS.
- protobuf
- IO libraries hdf5, leveldb, snappy, lmdb
None outside of C++ compiler and make
- CUDA 8 or 7.5 is recommended for GPU mode.
- Cmake > 3
- Bazel
By default DeepDetect automatically relies on a modified version of Caffe, https://github.com/beniz/caffe/tree/master This version includes many improvements over the original Caffe, such as sparse input data support, exception handling, class weights, object detection, and various additional losses and layers.
The code makes use of C++ policy design for modularity, performance and putting the maximum burden on the checks at compile time. The implementation uses many features from C++11.
-
Image classification Web interface: HTML and javascript classification image demo in demo/imgdetect
-
Image similarity search: Python script for indexing and searching images is in demo/imgsearch
-
Image object detection: Python script for object detection within images is in demo/objdetect
- List of examples, from MLP for data, text, multi-target regression to CNN and GoogleNet, finetuning, etc...: http://www.deepdetect.com/overview/examples/
Caffe | Tensorflow | Source | Top-1 Accuracy (ImageNet) | |
---|---|---|---|---|
AlexNet | Y | N | BVLC | 57.1% |
SqueezeNet | Y | N | DeepScale | 59.5% |
Inception v1 / GoogleNet | Y | Y | BVLC / Google | 67.9% |
Inception v2 | N | Y | 72.2% | |
Inception v3 | N | Y | 76.9% | |
Inception v4 | N | Y | 80.2% | |
ResNet 50 | Y | Y | MSR | 75.3% |
ResNet 101 | Y | Y | MSR | 76.4% |
ResNet 152 | Y | Y | MSR | 77% |
Inception-ResNet-v2 | N | Y | 79.79% | |
VGG-16 | Y | Y | Oxford | 70.5% |
VGG-19 | Y | Y | Oxford | 71.3% |
ResNext 50 | Y | N | https://github.com/terrychenism/ResNeXt | 76.9% |
ResNext 101 | Y | N | https://github.com/terrychenism/ResNeXt | 77.9% |
ResNext 152 | Y | N | https://github.com/terrychenism/ResNeXt | 78.7% |
DenseNet-121 | Y | N | https://github.com/shicai/DenseNet-Caffe | 74.9% |
DenseNet-161 | Y | N | https://github.com/shicai/DenseNet-Caffe | 77.6% |
DenseNet-169 | Y | N | https://github.com/shicai/DenseNet-Caffe | 76.1% |
DenseNet-201 | Y | N | https://github.com/shicai/DenseNet-Caffe | 77.3% |
VOC0712 (object detection) | Y | N | https://github.com/weiliu89/caffe/tree/ssd | 71.2 mAP |
More models:
- List of free, even for commercial use, deep neural nets for image classification, and character-based convolutional nets for text classification: http://www.deepdetect.com/applications/list_models/
DeepDetect comes with a built-in system of neural network templates (Caffe backend only at the moment). This allows the creation of custom networks based on recognized architectures, for images, text and data, and with much simplicity.
Usage:
- specify
template
to use, frommlp
,convnet
andresnet
- specify the architecture with the
layers
parameter:- for
mlp
, e.g.[300,100,10]
- for
convnet
, e.g.["1CR64","1CR128","2CR256","1024","512"], where the main pattern is
xCRywhere
yis the number of outputs (feature maps),
CRstands for Convolution + Activation (with
reluas default), and
xspecifies the number of chained
CRblocks without pooling. Pooling is applied between all
xCRy`
- for
- for
resnets
:- with images, e.g.
["Res50"]
where the main pattern isResX
with X the depth of the Resnet - with character-based models (text), use the
xCRy
pattern of convnets instead, with the main difference thatx
now specifies the number of chainedCR
blocks within a resnet block - for Resnets applied to CSV or SVM (sparse data), use the
mlp
pattern. In this latter case, at the moment, theresnet
is built with blocks made of two layers for each specified layer after the first one. Here is an example:[300,100,10]
means that a first hidden layer of size300
is applied followed by aresnet
block made of two100
fully connected layer, and another block of two10
fully connected layers. This is subjected to future changes and more control.
- with images, e.g.
DeepDetect is designed and implemented by Emmanuel Benazera [email protected].
Below are instructions for Ubuntu 14.04 LTS. For other Linux and Unix systems, steps may differ, CUDA, Caffe and other libraries may prove difficult to setup. If you are building on 16.04 LTS, look at https://github.com/beniz/deepdetect/issues/126 that tells you how to proceed.
Beware of dependencies, typically on Debian/Ubuntu Linux, do:
sudo apt-get install build-essential libgoogle-glog-dev libgflags-dev libeigen3-dev libopencv-dev libcppnetlib-dev libboost-dev libboost-iostreams-dev libcurlpp-dev libcurl4-openssl-dev protobuf-compiler libopenblas-dev libhdf5-dev libprotobuf-dev libleveldb-dev libsnappy-dev liblmdb-dev libutfcpp-dev cmake libgoogle-perftools-dev unzip
For compiling along with Caffe:
mkdir build
cd build
cmake ..
make
If you are building for one or more GPUs, you may need to add CUDA to your ld path:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
If you would like to build with cuDNN, your cmake
line should be:
cmake .. -DUSE_CUDNN=ON
To target the build of underlying Caffe to a specific CUDA architecture (e.g. Pascal), you can use:
cmake .. -DCUDA_ARCH="-gencode arch=compute_61,code=sm_61"
If you would like a CPU only build, use:
cmake .. -DUSE_CPU_ONLY=ON
If you would like to constrain Caffe to CPU only, use:
cmake .. -DUSE_CAFFE_CPU_ONLY=ON
If you would like to build with XGBoost, include the -DUSE_XGBOOST=ON
parameter to cmake
:
cmake .. -DUSE_XGBOOST=ON
If you would like to build the GPU support for XGBoost (experimental from DMLC), use the -DUSE_XGBOOST_GPU=ON
parameter to cmake
:
cmake .. -DUSE_XGBOOST=ON -DUSE_XGBOOST_GPU=ON
First you must install Bazel and Cmake with version > 3.
And other dependencies:
sudo apt-get install python-numpy swig python-dev python-wheel unzip
If you would like to build with Tensorflow, include the -DUSE_TF=ON
paramter to cmake
:
cmake .. -DUSE_TF=ON
If you would like to constrain Tensorflow to CPU, use:
cmake .. -DUSE_TF=ON -DUSE_TF_CPU_ONLY=ON
You can combine with XGBoost support with:
cmake .. -DUSE_TF=ON -DUSE_XGBOOST=ON
Simply specify the option via cmake command line:
cmake .. -DUSE_TSNE=ON
Note: running tests requires the automated download of ~75Mb of datasets, and computations may take around thirty minutes on a CPU-only machines.
To prepare for tests, compile with:
cmake -DBUILD_TESTS=ON ..
make
Run tests with:
ctest
cd build/main
./dede
DeepDetect [ commit 73d4e638498d51254862572fe577a21ab8de2ef1 ]
Running DeepDetect HTTP server on localhost:8080
Main options are:
-host
to select which host to run on, default islocalhost
, use0.0.0.0
to listen on all interfaces-port
to select which port to listen to, default is8080
-nthreads
to select the number of HTTP threads, default is10
To see all options, do:
./dede --help
To use deepdetect without the client/server architecture while passing the exact same JSON messages from the API:
./dede --jsonapi 1 <other options>
where <other options>
stands for the command line parameters from the command line JSON API:
-info (/info JSON call) type: bool default: false
-service_create (/service/service_name call JSON string) type: string default: ""
-service_delete (/service/service_name DELETE call JSON string) type: string default: ""
-service_name (service name string for JSON call /service/service_name) type: string default: ""
-service_predict (/predict POST call JSON string) type: string default: ""
-service_train (/train POST call JSON string) type: string default: ""
-service_train_delete (/train DELETE call JSON string) type: string default: ""
-service_train_status (/train GET call JSON string) type: string default: ""
The options above can be obtained from running
./dede --help
Example of creating a service then listing it:
./dede --jsonapi 1 --service_name test --service_create '{"mllib":"caffe","description":"classification service","type":"supervised","parameters":{"input":{"connector":"image"},"mllib":{"template":"googlenet","nclasses":10}},"model":{"templates":"/path/to/deepdetect/templates/caffe/","repository":"/path/to/model/"}}'
Note that in command line mode the --service_xxx
calls are executed sequentially, and synchronously. Also note the logs are those from the server, the JSON API response is not available in pure command line mode.
See tutorials from http://www.deepdetect.com/tutorials/tutorials/
- DeepDetect (http://www.deepdetect.com/)
- Caffe (https://github.com/BVLC/caffe)
- XGBoost (https://github.com/dmlc/xgboost)
- T-SNE (https://github.com/DmitryUlyanov/Multicore-TSNE)