V1.0

Author: CasiaFan

.gitignore

@@ -0,0 +1 @@
+cmake-build-debug

.idea/codeStyles/Project.xml

@@ -0,0 +1,40 @@
+cmake_minimum_required(VERSION 3.7)
+#project(TF_CPP_DETECT)
+
+set(CMAKE_CXX_STANDARD 11)
+ADD_COMPILE_OPTIONS(-fPIC)
+
+set(SOURCE_FILES detect.cpp main.cpp base64.cpp base64.h)
+add_executable(tf_detect_crow ${SOURCE_FILES})
+SET_TARGET_PROPERTIES(tf_detect_crow PROPERTIES PREFIX "")
+
+# crow lib
+#include_directories("/usr/local/include")
+
+# OpenCV libs
+find_package(OpenCV REQUIRED)
+include_directories(${OpenCV_INCLUDE_DIRS})
+target_link_libraries(tf_detect_crow ${OpenCV_LIBS})
+
+# boost lib
+find_package(Boost REQUIRED COMPONENTS system thread)
+include_directories( ${Boost_INCLUDE_DIRS} )
+target_link_libraries(tf_detect_crow ${Boost_SYSTEM_LIBRARY} ${Boost_THREAD_LIBRARY})
+
+# pthread link
+target_link_libraries(tf_detect_crow -lpthread -lm)
+# ==================== PATHS TO SPECIFY! ==================== #
+
+# Eigen lib headers
+include_directories("/usr/local/include/eigen3")
+include_directories("/usr/local/include/google")
+
+# TensorFlow headers
+include_directories("/usr/local/include/tf/")
+include_directories("/usr/local/include/tf/bazel-genfiles/")
+include_directories("/usr/local/include/tf/tensorflow/")
+include_directories("/usr/local/include/tf/third-party/")
+
+# Link TensorFlow libs
+target_link_libraries(tf_detect_crow "/usr/local/lib/libtensorflow_cc.so")
+target_link_libraries(tf_detect_crow "/usr/local/lib/libtensorflow_framework.so")

.idea/misc.xml

@@ -0,0 +1,55 @@
+This is an example how to run a TensorFlow object detection model as web server in TensorFlow C++ API.
+
+### Brief Introduction
+1). Construct a class for TF model like this:
+```c++
+class Detector {
+    std::unique_ptr<tensorflow::Session> session;
+    public:
+        int loadModel();
+        int detect();
+};
+``` 
+First, `loadModel` to initialize and load graph into session, then use 
+session in `detect` for prediction.
+
+2). use crow the start a web server in `main` (crow is inspired by python `FLASK`. 
+If you are familiar with FLASK, crow is easy to use.)
+
+### Requirements
+- [TensorFlow 1.8.0rc1](https://github.com/tensorflow)
+- [OpenCV 3.4.0](https://opencv.org/releases.html)
+- Boost 1.5.8
+- Ubuntu 16.04
+
+### Install 
+**Install TensorFlow C++ and OpenCV**: see this [blog](https://medium.com/@fanzongshaoxing/use-tensorflow-c-api-with-opencv3-bacb83ca5683)
+
+**Install Boost** <br>
+```bash
+sudo apt-get install libboost-all-dev
+```
+
+### Usage
+1. compile the project
+```bash
+cmake .
+make
+```
+2. run tf-cpp web service
+```bash
+./tf_detect_crow
+```
+
+3. test with python script
+``` 
+python test_cpp_api.py
+```
+
+### Acknowledgement
+Great appreciation to following project and code snippet: 
+
+- [opencv_tensor.cc](https://gist.github.com/kyrs/9adf86366e9e4f04addb)
+- [tensorflow object detection cpp](https://github.com/lysukhin/tensorflow-object-detection-cpp)
+- [crow example](https://github.com/jolks/crow-template/blob/master/src/example.cpp)
+- [Encoding and decoding base 64 with c++](https://renenyffenegger.ch/notes/development/Base64/Encoding-and-decoding-base-64-with-cpp)

.idea/modules.xml

@@ -0,0 +1,95 @@
+//
+// Code source: https://renenyffenegger.ch/notes/development/Base64/Encoding-and-decoding-base-64-with-cpp
+//
+
+#include "base64.h"
+#include <iostream>
+
+static const std::string base64_chars =
+             "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+             "abcdefghijklmnopqrstuvwxyz"
+             "0123456789+/";
+
+
+static inline bool is_base64(unsigned char c) {
+  return (isalnum(c) || (c == '+') || (c == '/'));
+}
+
+std::string base64_encode(unsigned char const* bytes_to_encode, unsigned int in_len) {
+  std::string ret;
+  int i = 0;
+  int j = 0;
+  unsigned char char_array_3[3];
+  unsigned char char_array_4[4];
+
+  while (in_len--) {
+    char_array_3[i++] = *(bytes_to_encode++);
+    if (i == 3) {
+      char_array_4[0] = (char_array_3[0] & 0xfc) >> 2;
+      char_array_4[1] = ((char_array_3[0] & 0x03) << 4) + ((char_array_3[1] & 0xf0) >> 4);
+      char_array_4[2] = ((char_array_3[1] & 0x0f) << 2) + ((char_array_3[2] & 0xc0) >> 6);
+      char_array_4[3] = char_array_3[2] & 0x3f;
+
+      for(i = 0; (i <4) ; i++)
+        ret += base64_chars[char_array_4[i]];
+      i = 0;
+    }
+  }
+
+  if (i)
+  {
+    for(j = i; j < 3; j++)
+      char_array_3[j] = '\0';
+
+    char_array_4[0] = ( char_array_3[0] & 0xfc) >> 2;
+    char_array_4[1] = ((char_array_3[0] & 0x03) << 4) + ((char_array_3[1] & 0xf0) >> 4);
+    char_array_4[2] = ((char_array_3[1] & 0x0f) << 2) + ((char_array_3[2] & 0xc0) >> 6);
+
+    for (j = 0; (j < i + 1); j++)
+      ret += base64_chars[char_array_4[j]];
+
+    while((i++ < 3))
+      ret += '=';
+
+  }
+
+  return ret;
+
+}
+
+std::string base64_decode(std::string const& encoded_string) {
+  int in_len = encoded_string.size();
+  int i = 0;
+  int j = 0;
+  int in_ = 0;
+  unsigned char char_array_4[4], char_array_3[3];
+  std::string ret;
+
+  while (in_len-- && ( encoded_string[in_] != '=') && is_base64(encoded_string[in_])) {
+    char_array_4[i++] = encoded_string[in_]; in_++;
+    if (i ==4) {
+      for (i = 0; i <4; i++)
+        char_array_4[i] = base64_chars.find(char_array_4[i]);
+
+      char_array_3[0] = ( char_array_4[0] << 2       ) + ((char_array_4[1] & 0x30) >> 4);
+      char_array_3[1] = ((char_array_4[1] & 0xf) << 4) + ((char_array_4[2] & 0x3c) >> 2);
+      char_array_3[2] = ((char_array_4[2] & 0x3) << 6) +   char_array_4[3];
+
+      for (i = 0; (i < 3); i++)
+        ret += char_array_3[i];
+      i = 0;
+    }
+  }
+
+  if (i) {
+    for (j = 0; j < i; j++)
+      char_array_4[j] = base64_chars.find(char_array_4[j]);
+
+    char_array_3[0] = (char_array_4[0] << 2) + ((char_array_4[1] & 0x30) >> 4);
+    char_array_3[1] = ((char_array_4[1] & 0xf) << 4) + ((char_array_4[2] & 0x3c) >> 2);
+
+    for (j = 0; (j < i - 1); j++) ret += char_array_3[j];
+  }
+
+  return ret;
+}

.idea/tensorflow_c++_detection_for_web_service_example.iml

@@ -0,0 +1,13 @@
+//
+// Code source: https://renenyffenegger.ch/notes/development/Base64/Encoding-and-decoding-base-64-with-cpp
+
+//
+
+#ifndef PROJECT_BASE64_H
+#define PROJECT_BASE64_H
+#include <string>
+
+std::string base64_encode(unsigned char const* , unsigned int len);
+std::string base64_decode(std::string const& s);
+
+#endif //PROJECT_BASE64_H