test_fight.cpp

#include <chrono>
#include <fstream>
#include <iostream>
#include <sstream>
#include <string>
#include <vector>
#include <assert.h>
#include <thread>
#include <stdio.h>
#include "libai_core.hpp"
#include "config.hpp"

using namespace std;
using namespace ucloud;

/**
 * ./test_fight
 */
int main(int argc, char* argv[]) {
    float threshold, nms_threshold;
    AlgoAPIName mainApiName, subApiName;
    ucloud::TvaiResolution maxTarget={0,0};
    ucloud::TvaiResolution minTarget={0,0};
    std::vector<ucloud::TvaiRect> pRoi;
    int use_batch = 0;
    task_parser(TASKNAME::FIGHT, threshold,nms_threshold, mainApiName, subApiName, use_batch);
    AlgoAPISPtr algoHandle = AICoreFactory::getAlgoAPI(mainApiName);
    RET_CODE ret = algoHandle->init(modelInfo[mainApiName]);
    cout << "@@ init return = " << ret << endl;
    ret = algoHandle->set_param(threshold, nms_threshold, maxTarget, minTarget, pRoi);

    std::string datapath = "fight/";
    std::ifstream infile;
    std::string filename = datapath + "list.txt";

    //DATA
    infile.open(filename, std::ios::in);
    std::string imgname;
    std::vector<string> vec_imgnames;
    bool use_yuv = false;
    while(infile >> imgname){
        if(imgname.find(".yuv")>=0 && imgname.find(".yuv")!=std::string::npos){
            use_yuv = true;
        }
        vec_imgnames.push_back(imgname);
    }

    double total_tm_cost = 0;
    BatchImageIN batch_tvimages;
    const int batch_size_in = 8;
    for(auto iter=vec_imgnames.begin(); iter!=vec_imgnames.end(); iter++){
        unsigned char* ptrImg = nullptr;
        unsigned char* ptrShow = nullptr;
        int width, height, stride;
        int _width, _height, _stride;
        int inpSz;
        TvaiImageFormat inpFmt = TVAI_IMAGE_FORMAT_NV21;
        std::string imgname_full = datapath + *iter;
        if(!use_yuv){
            ptrImg = readImg_to_NV21(imgname_full, width, height, stride );
            ptrShow = readImg(imgname_full, _width, _height);
            _stride = _width;
        }
        else{
            width = 1920; height = 1080; stride = width;
            ptrImg = yuv_reader(imgname_full, width, height);
        }
        inpSz = 3*stride*height/2*sizeof(unsigned char);
        TvaiImage inpImg(inpFmt,width,height,stride,ptrImg,inpSz);
        batch_tvimages.push_back(inpImg);
        if(batch_tvimages.size()==batch_size_in){
            //ALGO
            BatchBBoxIN batch_bboxes;
            VecObjBBox result;
            ret = algoHandle->run(batch_tvimages, result);
            cout << "@@ run return = " << ret << endl;
            if(!result.empty()){
                cout << "fight prob = " << result[0].confidence << endl;
            }
            for(auto iter=batch_tvimages.begin(); iter!=batch_tvimages.end(); iter++ ){
                free(iter->pData);
            }
            batch_tvimages.clear();
        }
    }

    for(auto iter=batch_tvimages.begin(); iter!=batch_tvimages.end(); iter++ ){
        free(iter->pData);
    }

}