update

zhangmeng

2019-11-15 4d025c4a8c81c0bfdd159586528db0af72e6a6d6

 csdk.cpp

@@ -12,103 +12,106 @@

using namespace csdk_wrap;

static sdkface *face = NULL;
static sdkyolo *yolo = NULL;
static VecFunc dtors_;

int c_api_face_detector_init(const int tm, const int gi, const int minFaces, const int rollAngle){
    if (!face) face = new sdkface();
    return face->init_face_detector(tm, gi, minFaces, rollAngle);
    return init_face_detector(tm, gi, minFaces, rollAngle, dtors_);
}

int c_api_face_property_init(const int tm){
    if (!face) face = new sdkface();
    return face->init_face_property(tm);
    return init_face_property(tm, dtors_);
}

int c_api_face_extractor_init(const int tm, const int gi){
    if (!face) face = new sdkface();
    return face->init_face_extractor(tm, gi);
    return init_face_extractor(tm, gi, dtors_);
}

int c_api_face_tracker_init(const int tm, const int gi, const int wid, const int hei,
                              const int maxFaces, const int detinterval, const int sampleSize){
    if (!face) face = new sdkface();
    return face->init_face_tracker(tm, gi, wid, hei, maxFaces, detinterval, sampleSize);

   return init_face_tracker(tm, gi, wid, hei, maxFaces, detinterval, sampleSize, dtors_);
}

int c_api_face_track_resize(const int chan, const int wid, const int hei){
    return face_track_resize(chan, wid, hei);
}

YoloHandle c_api_yolo_init(
    const char *fcfg, const char *fweights, const char *fname,
    const int gpu_index){
    
    if (!yolo) yolo = new sdkyolo;
    return yolo->init_yolo(fcfg, fweights, fname, gpu_index);    

    return init_yolo_detector(fcfg, fweights, fname, gpu_index, dtors_);
}

void c_api_release(){
    if (face) delete face;
    if (yolo) delete yolo;
    for(auto &i : dtors_){
        i();
    }
    dtors_.clear();
}

////////////////////////////////////////////////

cFacePos* c_api_face_detect(int *faceCount, uchar*data, const int w, const int h, const int channel){
    if (!face) return NULL;
    
    const cIMAGE img{data, w, h, 3};
    cFacePos *fpos = NULL;

    int ret = face->face_detect(&img, channel, (void**)&fpos, faceCount);

    if (ret <= 0) return NULL;
    return fpos;
    return face_detect(faceCount, &img, channel);
}

cThftResult c_api_face_property(const cFacePos* pos, uchar*data, const int w, const int h, const int channel){
    if (!face) return cThftResult{-1,-1,-1,-1,-1};

    const cIMAGE img{data, w, h, 3};
    return face->face_property(*pos, &img, channel);
    return face_property(*pos, &img, channel);
}

uchar* c_api_face_extract(int *featLen, const cFacePos* pos, uchar*data, const int w, const int h, const int channel){
    if (!face) return NULL;

    const cIMAGE img{data, w, h, 3};
    uchar *feat = NULL;
    int ret = face->face_extract_feature(*pos, &img, channel, (void**)&feat, featLen);
    if (ret <= 0) return NULL;
    return feat;
    return face_extract_feature(featLen, *pos, &img, channel);
}

float c_api_face_compare(uchar *feat1, uchar *feat2){
    if (!face) return NULL;
    return face->face_compare(feat1, feat2);
    return face_compare(feat1, feat2);
}

cRECT* c_api_face_track_only(int *fCount, uchar *data, const int wid, const int hei, const int channel){
    const cIMAGE img{data, wid, hei, 3};

    return face_track_only(fCount, &img, channel);
}

cFaceInfo* c_api_face_track_detect(int *fCount, uchar *data, const int wid, const int hei, const int channel){
    const cIMAGE img{data, wid, hei, 3};

    return face_track_detect(fCount, &img, channel);
}

cFaceInfo* c_api_face_track(int *fCount, uchar *data, const int wid, const int hei, const int channel){
    if (!face) return NULL;
    const cIMAGE img{data, wid, hei, 3};

    cFaceInfo *info = NULL;
    int ret = face->face_track(&img, channel, (void**)&info, fCount);
    if (ret <= 0) return NULL;
    return info;
    return face_track(fCount, &img, channel);
}


/// yolo api
cObjInfo* c_api_yolo_detect(YoloHandle handle, int *objCount, uchar*data, const int w, const int h, const float thrsh, const int use_means){
    if (!yolo) return NULL;

    const cIMAGE img{data, w, h, 3};
    cObjInfo *info = NULL;
    int ret = yolo->yolo_detect(handle, &img, thrsh, use_means, (void**)&info, objCount);
    if (ret <= 0) return NULL;
    return info;
    return yolo_detect(handle, objCount, &img, thrsh, use_means);
}

const char* c_api_yolo_obj_name(const int typ){
    if (!yolo) return NULL;
    return yolo_obj_name_by_type(typ);
}

    return yolo->yolo_obj_name_by_type(typ);
// plateid api
int c_api_plate_id_init(const cPlateIDCfg *config, char *soPath) {
    return init_plate_id_detector(config, soPath);
}

cPlateIDResult* c_api_plate_id_detect(int *plateIDCount, uchar *data, const int w, const int h) {
    const cIMAGE img{data, w, h, 3};
    return plate_id_detect(plateIDCount, &img);
}

int c_api_plate_id_free() {
    return uninit_plate_id_detector();
}