interface.h

Go to the documentation of this file.

//
// Created by Jinghui Zhou on 2023/1/4.
//
 
#ifndef AISDEPLOYC_INTERFACE_H
#define AISDEPLOYC_INTERFACE_H
 
const std::string AisDeployCVersion="v0.3.3b";
 
#ifdef DEPLOY_ON_WINDOWS
#define AisDeployC_API extern "C" __declspec(dllexport)
#else
#define AisDeployC_API extern "C"
#endif
 
 
typedef struct bounding_box
{
 
    std::string category = "";        // category name, get from config file
    float score = 0.;                  // confidence
    float xmin = 0;                   // target bbox position, left top x
    float ymin = 0;                   // target bbox position, left top y
    float xmax = 0;                   // target bbox position, right bottom x
    float ymax = 0;                   // target bbox position, right bottom y
 
}bounding_box;
 
typedef struct classification
{
 
    std::string category = "";        // category name, get from config file
    float score = 0.;                  // confidence
 
}classification;
 
AisDeployC_API int check_version();
 
AisDeployC_API std::string get_version();
AisDeployC_API void *initialize(
        const char *model_path,
        int gpu_id,
        int *state
        );
 
AisDeployC_API int process_images(
        void *base,
        std::vector<cv::Mat> & imgs
        );
 
AisDeployC_API int get_categories(
        void *base,
        std::vector<std::string> & out_categories
);
 
AisDeployC_API int get_per_category_binary_masks(
        void *base,
        std::vector<std::vector<cv::Mat> > & per_category_masks
);
 
AisDeployC_API int get_per_category_score_maps(
        void *base,
        std::vector<std::vector<cv::Mat> > & per_category_score_maps
);
 
AisDeployC_API int get_binary_mask_from_score_map(
        cv::Mat score_map,
        cv::Mat & binary_mask,
        float thresh
);
 
 
AisDeployC_API int erode_binary_mask(
        cv::Mat & binary_mask,
        int kernel_size,
        int iter_erode
);
 
AisDeployC_API int dilate_binary_mask(
        cv::Mat & binary_mask,
        int kernel_size,
        int iter_dilate
);
 
 
AisDeployC_API int cal_per_category_instances(
        void *base,
        std::vector<std::vector<cv::Mat> >  per_category_score_maps,
        std::vector<std::vector<cv::Mat> > per_category_masks
);
 
AisDeployC_API int get_per_batch_instance_number(
        void *base,
        std::vector<int> & per_batch_instance_number
);
 
AisDeployC_API  int get_per_batch_instances(
        void *base,
        std::vector<std::vector<bounding_box>> & per_batch_instances
);
 
 
 
AisDeployC_API  int get_per_batch_instance_mask_areas(
        void *base,
        std::vector<std::vector<float>> & per_batch_instance_mask_areas
);
 
AisDeployC_API int release(
        void *base
);
 
 
AisDeployC_API int update_license(void *base, const char* path);
 
AisDeployC_API int generate_license(void *base);
 
AisDeployC_API int py_process_json_str(void *base, const char *input, int input_size);
 
AisDeployC_API int py_get_json_str_results(void *base, char **output, int *output_size);
 
AisDeployC_API int py_free_result(char *output);
 
AisDeployC_API  int get_per_batch_embeddings(
        void *base,
        std::vector<std::vector<float>> & per_batch_embeddings
);
 
AisDeployC_API  int get_per_batch_embeddings_3d(
        void *base,
        std::vector<std::vector<std::vector<std::vector<float>>>> & per_batch_embeddings
);
 
AisDeployC_API int process_embedding_3d_prompt(
        void *base,
        std::vector<std::vector<std::vector<std::vector<float>>>>  batch_embeddings,
        std::vector<std::vector<std::vector<float>>> point_coords,
        std::vector<std::vector<float>> point_labels,
        std::vector<std::vector<std::vector<std::vector<float>>>> batch_mask_input,
        std::vector<float> has_mask_input,
        std::vector<std::vector<float>> orig_im_size
);
 
AisDeployC_API  int load_keys_embeddings(
        void *base,
        std::vector<std::string> & vec_keys,
        std::vector<std::vector<float>> & vec_embeddings
);
 
 
AisDeployC_API  int compare_with_ground_embeddings(
        void *base,
        std::vector<std::vector<float>> & vec_embeddings,
        std::vector<std::string> & ground_keys,
        std::vector<std::vector<float>> & vec_scores
);
 
 
AisDeployC_API int py_compare_with_ground_embeddings(void *base, const char *input, int input_size, char **output, int *output_size);
 
AisDeployC_API int py_process_decoder(void *base, const char *input, int input_size, char **output, int *output_size);
 
 
AisDeployC_API int py_get_json_str_results(void *base, char **output, int *output_size);
 
 
AisDeployC_API  int get_per_batch_classification(
        void *base,
        std::vector<classification> & per_batch_cls
);
 
AisDeployC_API  int get_per_batch_pose_estimation(
        void *base,
        std::vector<std::vector<std::vector<float>>>  & per_batch_pose
);
 
#endif //AISDEPLOYC_INTERFACE_H