Simple Cloud-AI (SCAI) is the simple api server for AI image recognition. The SCAI now provides two types of API, that are image classification and object detection.
The API demo portal :
Before establish two types of cloud AI APIs. You have to make sure of installing the following component.
- python 3.x (better >= 3.5.x)
- flask package (better >= 0.12.2)
- numpy package (better >= 1.14.2)
- matplotlib package (better >= 1.5.3)
- tensorflow package (better >=1.7.0)
- PIL package (pillow, better >= 5.1.0)
- cv2 package (opencv, better >= 3.4.0)
- git (version control)
- git clone https://github.com/tensorflow/tensorflow.git
- git clone https://github.com/tensorflow/models.git
We assume you have already git clone the two main repository under path /home/user/tensorflow and home/user/models and install the tensorflow package in python (simply $ pip install tensorflow or $ pip install tensorflow-gpu).
The following is the basic API server structure.
+ .git
+ object_detection (you have to git clone tensorflow/models first, see below)
+ core
+ protos
+ utils
- __init__.py
+ session (you have to create it for conserving the task session)
+ static (for front-end js or css resource)
+ templates (for portal pages or api accessing)
+ usr (for ai models and label file, and also conserving config file for api settings)
- .gitignore
- index.py (flask main entry)
- label_image.py (modified from tensorflow for image classification core)
- object_detection.py (modified from tensorflow for object detection core)
- README.md
The above is the recommanded structure for running scai.
The image classification is the issue for classifying the image into classes. The output would be the probability standing for how much the image is likely belonging to the specific class. The following is the demo for recognizing a flower image.
Here we simply demo how to use the resource from google tensorflow to train your own dataset.
- Retrain the model
You have to download retrain.py from https://github.com/tensorflow/hub/tree/master/examples/image_retraining first, and put it to tensorflow/tensorflow/examples/image_retraining/.
You can run the commad like below.
# the path you git clone tensorflow
cd /home/user/tensorflow/tensorflow/examples/image_retraining/
python retrain.py \
--bottleneck_dir=bottleneck \
--how_many_training_step 10000 \
--model_dir=inception-2015-12-05 \
--output_graph=retrained_graph.pb \
--output_labels=retrained_labels.txt \
--image_dir /path/to/your/datasetThe model-dir can be downloaded from http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz (for example), and uncompressed it to the same path with retrain.py. You can also change other inception version from tensorflow.org (or use --tfhub_module to choose different modules).
The image_dir structure is simply like below.
# /path/to/your/dataset
+ class1
- image1.jpg
- image2.jpg
- ...
+ class2
- image1.jpg
- image2.jpg
- ...
+ class3
+ ...
The step is optional if you are going to deploy the model on the mobile or development board. After retain the model with your own dataset, the model can be further optimized.
cd /home/user/tensorflow/tensorflow/examples/image_retraining/
python -m tensorflow.python.tools.optimize_for_inference \
--input=retrained_graph.pb \
--output=retrained_graph_op.pb \
--input_names=module_apply_default/hub_input/Mul \
--output_names=final_result \
--frozen_graph=TrueThe input is output of Train your own dataset on the first step.
The input_names could be changed from time to time, you can simple use the following script to find Mul under proper scope name.
def listAllOps(PB_PATH):
with tf.Session() as sess:
with tf.gfile.FastGFile(PB_PATH, 'rb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
sess.graph.as_default()
tf.import_graph_def(graph_def, name='tfgraph')
for op in sess.graph.get_operations():
print("{}".format(op.name))
# for exmpale:
# tfgraph/module_apply_default/hub_input/Mul
listAllOps('retrained_graph.pb')After you retained the model, or even optimized the model, you can start the classification test.
python /home/user/tensorflow/tensorflow/examples/label_image/label_image.py \
--output_layer=final_result \
--labels=/home/user/tensorflow/tensorflow/examples/image_retraining/retrained_labels.txt \
--image=/path/to/your/dataset/test_image.jpg \
--graph=/home/user/tensorflow/tensorflow/examples/image_retraining/retrained_graph_op.pb \
--input_layer=module_apply_default/hub_input/MulHere we used a optimizated model (retrained_graph_op.pb) as the example.
The step is optional if you are going to deploy the model on the mobile or development board. After retained the model with your own dataset and optimized the retrained model, you can further quantizated the model (you can simply think of it as the data compression) into a small model.
python /home/user/tensorflow/tensorflow/tools/quantization/quantize_graph.py \
--input=/home/user/tensorflow/tensorflow/examples/image_retraining/retrained_graph_op.pb \
--output=/home/user/tensorflow/tensorflow/examples/image_retraining/retrained_op_rounded.pb \
--output_node_names=final_result \
--mode=eightbitYou would find it simply downsize the model. You can embed it to your resource limited device.
In google tensorflow, you can simply run the script to retrain and quantize the model once for all.
cd /home/user/tensorflow/tensorflow/examples/image_retraining/
python retrain.py \
--bottleneck_dir=bottleneck_mobilenet \
--how_many_training_step 10000 \
--image_dir /path/to/your/dataset \
--tfhub_module https://tfhub.dev/google/imagenet/mobilenet_v1_100_224/quantops/feature_vector/1 \
--output_graph=retrained_graph_mobilenet_v1.pb \
--output_labels=retrained_labels.txtYou can choose differenttfhub_module on the website https://www.tensorflow.org/hub/。
After you retrained model with your own dataset. You can simply put the model and label into /usr folder, and configurate the /usr/image_classification.config on the section default.
[default]
LABELS = retrained_labels.txt
GRAPH = retrained_graph_mobilenet_v1.pb
INPUT_LAYER = module_apply_default/hub_input/Mul
OUTPUT_LAYER = final_resultObject detection is the issue to recognize the specific objects on the image.
In SCAI, we integrates with lots of tensorflow components. If you want to use object detection API, you have to prepare necessary components (replicate from tensorflow/models) in the root path of scai.
cd scai
mkdir ./object_detection
# make sure you have already git clone the tensorflow/models
cp models/research/object_detection/__init__.py ./object_detection/
cp -r models/research/object_detection/core ./object_detection/
cp -r models/research/object_detection/protos ./object_detection/
cp -r models/research/object_detection/utils ./object_detection/+ scai
+ object_detection
+ core
+ protos
+ utils
- __init__.py
For more training details, including labeling, training datasets, model deployment, etc., please refer to https://github.com/jiankaiwang/TF_ObjectDetection_Flow.
There are two running modes, job and runtime. The job mode execute each session (or task) as the single process with complete execution flow, e.g. loading library, loading graph, loading image, executing image classification or object detection, ... , etc. The runtime mode (the default mode) takes the each session as the part in the single process. It would be running on the background and watch the folder change over several seconds (--checktimeperoid). If there is file (named with the session name) created onto the folder, it would load the session image and directly start image classification or object detection without loading libraries or graph. The runtime mode would save about half execution time. You can simply use flag --runningmode to switch between modes.
The following are basic commands running the scai portal with its image classification and object detection modules.
git clone https://github.com/jiankaiwang/scai.git
cd /home/user/scai
# run image classification on the background
python label_image.py > /dev/null &2>&1
# run object detection on the background
python object_detection.py > /dev/null &2>&1
# run the main service portal
sudo python index.py The following we introduce how to establish scai service. Here we use nginx and uwsgi to run the scai API portal. We also establish two services for image classification and object detection, and create two processes of each of them on th background.
- Establish SCAI API service.
Install nginx.
## Replace $release with your corresponding Ubuntu release.
# deb http://nginx.org/packages/ubuntu/ $release nginx
# deb-src http://nginx.org/packages/ubuntu/ $release nginx
$ deb http://nginx.org/packages/ubuntu/ xenial nginx
$ deb-src http://nginx.org/packages/ubuntu/ xenial nginx
$ sudo apt-get update
$ sudo apt-get install -y nginxConfigurate the nginx.
$ sudo vim /etc/nginx/nginx.conf# add the command to http section
# upload body size
client_max_body_size 20m;$ sudo systemctl restart nginx.serviceInstall uwsgi.
$ sudo apt-get install build-essential python-dev
$ sudo pip install uwsgiClone the repository.
$ git clone https://github.com/jiankaiwang/scai.git
$ cd /home/user/scaiThe default run.py script is like the below.
from index import app as application
if __name__ == "__main__":
application.run()You can simply start the service by the command below.
$ uwsgi --http :5001 --wsgi-file ./run.py --enable-threadsCreate the ini for uwsgi.
$ vim /home/user/scai/scai.ini[uwsgi]
module = run:application
master = true
processes = 3
chdir = /home/user/scai/
socket = /tmp/scai.sock
logto = /tmp/scai.log
chmod-socket = 666
vacuum = true
http-socket = :5001
enable-threads = true
# ~ 50 MB
log-maxsize = 50000000
# allow send_file
wsgi-disable-file-wrapper = trueRun the uwsgi app.
$ uwsgi --ini /home/user/scai/scai.iniAdd soft link to the /usr/bin.
$ sudo ln -s /home/user/.local/bin/uwsgi /usr/bin/uwsgiEstablish the service.
$ sudo vim /etc/systemd/system/scai.service[Unit]
Description=Simple Cloud-AI Portal
After=network.target
[Service]
User=user
Group=user
ExecStart=/usr/bin/uwsgi --ini /home/user/scai/scai.ini
Restart=always
WorkingDirectory=/home/user/scai
[Install]
WantedBy=multi-user.target$ sudo systemctl status scai.service
$ sudo systemctl start scai.service
$ sudo systemctl enable scai.serviceAccess the portal over Nginx.
$ sudo vim /etc/nginx/sites-available/defaultserver {
listen 80;
server_name domain.example.com;
add_header "Access-Control-Allow-Origin" "*";
add_header "Access-Control-Allow-Methods" "GET, POST, OPTIONS, HEAD";
add_header "Access-Control-Allow-Headers" "Authorization, Origin, X-Requested-With, Content-Type, Accept";
location / {
include uwsgi_params;
uwsgi_pass unix:/tmp/scai.sock;
}
# let's encrypt
location ~ /.well-known {
root /home/user/;
allow all;
}
}It is not ideal to add scai subpath to other web. But if you have to integrate scai api to other systems, the alternative configuration is referred to the below.
{
...
add_header "Access-Control-Allow-Origin" "*";
add_header "Access-Control-Allow-Methods" "GET, POST, OPTIONS, HEAD";
add_header "Access-Control-Allow-Headers" "Authorization, Origin, X-Requested-With, Content-Type, Accept";
location ~ /imageclassification {
proxy_pass http://localhost:5001;
# Redefine the header fields that NGINX sends to the upstream server
proxy_set_header Host $host;
proxy_set_header X-Real-IP $remote_addr;
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
}
location ~ /objectdetection {
proxy_pass http://localhost:5001;
# Redefine the header fields that NGINX sends to the upstream server
proxy_set_header Host $host;
proxy_set_header X-Real-IP $remote_addr;
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
}
...
}- starting image classification module
$ sudo vim /etc/systemd/system/scai_ic.service[Unit]
Description=Simple Cloud-AI Image Classification
After=network.target
[Service]
User=user
Group=user
ExecStart=/usr/bin/python /home/user/scai/label_image.py --checktimeperoid=1
Restart=always
WorkingDirectory=/home/user/scai
[Install]
WantedBy=multi-user.target$ sudo systemctl status scai_ic.service
$ sudo systemctl start scai_ic.service
$ sudo systemctl enable scai_ic.service- starting object detection module
$ sudo vim /etc/systemd/system/scai_od.service[Unit]
Description=Simple Cloud-AI Object Detection
After=network.target
[Service]
User=user
Group=user
ExecStart=/usr/bin/python /home/user/scai/od.py --checktimeperoid=1
Restart=always
WorkingDirectory=/home/user/scai
[Install]
WantedBy=multi-user.target$ sudo systemctl status scai_od.service
$ sudo systemctl start scai_od.service
$ sudo systemctl enable scai_od.service- running session cleaning
You can set up cleaning job over crontab on linux. The following is the example to clean sessions which is processed two days before on every 4 hours.
$ sudo vim /etc/crontab30 */4 * * * user /home/user/scai/clean.py --sessionpath=/home/user/scai/session --keeptimeperoid=172800url # e.g. www.example.com, 192.168.1.2, etc.
/ # portal, ui demo webpage
/imageclassification ['GET','POST'] [return JSON] # upload image classification api
/imgclassres ['GET'] [return JSON] # get image classification result
/iciptimg ['GET'] [return IMAGE/PNG] # get upload image for classification
/objectdetection ['GET','POST'] [return JSON] # upload object detection
/odres ['GET'] [return JSON] # get object detection result
/odiptimg ['GET'] [return IMAGE/PNG] # get upload image for object detection
/odresimg ['GET'] [return IMAGE/PNG] # get result image of object detection