top of page
  • Jon Zhao

如何部署Intel OpenVINO的模型到OpenNCC

The key step of developing Edge-AI camera on open camera is to deploy deep learning neural network model at the edge. The model may be trained from Caffe *, tensorflow *, mxnet *, Kaldi *, paddy *, and onnx *.

OpenNCC SDK installation

Please install the SDK development environment according to documentation.Such as:

  • YOUR OPENNCC SDK INSTALL PATH'/Platform/Linux/readme.md

  • We use Linux for demo,other platform also well supported by OpenNCC.

If you still don't have a SDK,you could clone it from github openncc

  • To use Model Optimizer and BLOB Converter,please make sure the OpenVINO ToolKit already installed. You could download the OpenVINO here. And MUST choose the 2020 3 LTS version which comprehensive tested with OpenNCC.

  • If you want to download a model from Open Model Zoo,you need Model Downloader. Or you could also download from website.

Steps of deployment a deep learning model on edge

The following figure shows a complete AI model development process:


If we start with a existing model, the steps are simplified as follows:

Step1: Prepare a trained model

Use the Open Model Zoo to find open-source, pretrained, and preoptimized models ready for inference, or use your own deep-learning model.

Download a model from open model zoo,like:

  • $./downloader.py --name person-detection-retail-0002

  • Note: The OpenNCC support FP16 format,please use FP16 model.


Step2: Model Optimizer

Run the trained model through the Model Optimizer to convert the model to an Intermediate Representation (IR), which is represented in a pair of files (.xml and .bin). These files describe the network topology and contain the weights and biases binary data of the model.

  • If you download the model from Open model zoo,it is already IR files.So we don't need run model optimizer.

  • If you use your own model,you need run model optimizer,please follow Model Optimizer Developer Guide.

Step3: Converte to BLOB format

After the model optimization is completed,means you already have two files(.xml and .bin), the model needs to be converted to the Blob format before it can be deployed on OpenNCC.

You need run myriad_compile tool to packet the IR files to a BLOB file,the openncc using the blob file to inference the model.

  • Example:

$ ./myriad_compile -m input_xxx-fp16.xml -o output_xxx.blob -VPU_PLATFORM VPU_2480 -VPU_NUMBER_OF_SHAVES 8 -VPU_NUMBER_OF_CMX_SLICES 8

  • Note:

The myriad_compile is a tool of OpenVINO ToolKit,you need install the openvino on your development host. The tool under:

/opt/intel/openvino/deployment_tools/inference_engine/lib/intel64myriad_compile


Step4: Inference on OpenNCC and extract the results

Use the OpenNCC SDK to download the BLOB file,run inference and output results on OpenNCC Cameras.

The OpenNCC SDK would output two types of streams:

* Normal video stream,support YUV420,YUV422,MJPG,H.264 and H.265

* AI-Meta data stream is the binary results by inference with frame based data. The specific output structure depends on the model which running on OpenNCC.Take the person-detection-retail-0002 model as an example:


  • Outputs:

The net outputs "detection_output" blob with shape: [1x1xNx7], where N is the number of detected pedestrians. For each detection, the description has the format: [image_id, label, conf, x_min, y_min, x_max, y_max], where:

image_id - ID of image in batch

label - ID of predicted class

conf - Confidence for the predicted class

(x_min, y_min) - Coordinates of the top left bounding box corner

(x_max, y_max) - Coordinates of the bottom right bounding box corner.


So a AI-Meta frame you get from SDK would be this:



Demo and Running


1. Enter 'YOUR OPENNCC SDK INSTALL PATH'/Platform/Linux/Example/How_to/Load_a_model


2. Copy all the related files from sdk to this demo.


$ ./copy.sh  


3. Download the BLOB model file.


In the main.cpp,
 ........  
 //5.2 Image preprocessing parameter initialization  
    cam_info.inputDimWidth  = 300;  
    cam_info.inputDimHeight = 300;  
    cam_info.inputFormat    = IMG_FORMAT_BGR_PLANAR;  
    cam_info.meanValue[0]   = 0;  
    cam_info.meanValue[1]   = 0;  
    cam_info.meanValue[2]   = 0;  
    cam_info.stdValue       = 1;  
    // 5.2 Blob file path 
    const char *blob = "./blob/2020.3/object_classification/object_classification.blob";    

    //6. sdk initialization,and download the model to the openncc    
    ret = sdk_init(NULL, NULL, (char*) blob, &cam_info, sizeof(cam_info));  
    printf("sdk_init %d\n", ret);  
    if (ret < 0)  
        return -1;  


We need load the BLOB model file to the OpenNCC,if your OpenNCC version don't have a Flash or EMMC on board.


const char *blob = "./blob/2020.3/object_classification/object_classification.blob"; 

Here pass the BLOB model file path to the SDK,which you want to run on OpenNCC. You could change the blob file yourself.


4. Get the AI-meta data frame


     //Non-blocking read metedata data
        max_read_size = 512*1024;
        memset(recv_metedata, 0, max_read_size);
        if (read_meta_data(recv_metedata, &max_read_size, false) == 0)
        {
            memcpy(&hdr, recv_metedata, sizeof(frameSpecOut));
            printf("metehdr:type:%2d,seqNo:%-6d,size %d, NCC_T:(%dMS)\n", hdr.type, hdr.seqNo,hdr.size, hdr.res[0]);
        }  

5. Post-process to extract the results

5.1 Skip the openncc header


memedata = (char*) recv_metedata + sizeof(frameSpecOut)+16*sizeof(int);
obj_show_img_func(yuv420p, cameraCfg.camWidth, cameraCfg.camHeight,scale, src, 1, &cam_info,memedata , min_score);

5.2 Extract the results in obj_demo_show.cpp

    ....  
    #define MAX_INTEMS_RESULTS  200
    ....
    uint16_t* cls_ret = (uint16_t*)nnret; //nnret is the point of the ai-meta data
    ....
    for (i = 0; i < MAX_INTEMS_RESULTS; i++)
    {
     //Since the VPU output FP16 format,we need convert to fp32.
        int image_id = (int)(f16Tof32(cls_ret[i*7+0]));
        int label    = (int)(f16Tof32(cls_ret[i*7+1]));
        score =(float)f16Tof32(cls_ret[i*7+2]);

        if (image_id < 0) {
            break;
        }

        x0 = f16Tof32(cls_ret[i*7+3]);
        y0 = f16Tof32(cls_ret[i*7+4]);
        x1 = f16Tof32(cls_ret[i*7+5]);
        y1 = f16Tof32(cls_ret[i*7+6]);
     }
     
     



35 次查看0 則留言

最新文章

查看全部
bottom of page