[C API] Supplement C api for models (#1317)

* add batch_predict and initialized api for c

* add batch_predict and result str in C api

* modify visualization function

* add micro to add models batchify

* refactor and add models for ppdet others

* add base define

* fix

* refine predict api

c_api/fastdeploy_capi/vision/classification/ppcls/model.cc

@@ -38,21 +38,82 @@ FD_C_PaddleClasModelWrapper* FD_C_CreatePaddleClasModelWrapper(
 }
 
 void FD_C_DestroyPaddleClasModelWrapper(
-    __fd_take FD_C_PaddleClasModelWrapper* fd_c_paddleclas_model_wrapper) {
+    FD_C_PaddleClasModelWrapper* fd_c_paddleclas_model_wrapper) {
   delete fd_c_paddleclas_model_wrapper;
 }
 
 FD_C_Bool FD_C_PaddleClasModelWrapperPredict(
-    __fd_take FD_C_PaddleClasModelWrapper* fd_c_paddleclas_model_wrapper,
-    FD_C_Mat img, FD_C_ClassifyResultWrapper* fd_c_classify_result_wrapper) {
+    FD_C_PaddleClasModelWrapper* fd_c_paddleclas_model_wrapper, FD_C_Mat img,
+    FD_C_ClassifyResult* fd_c_classify_result) {
   cv::Mat* im = reinterpret_cast<cv::Mat*>(img);
   auto& paddleclas_model = CHECK_AND_CONVERT_FD_TYPE(
       PaddleClasModelWrapper, fd_c_paddleclas_model_wrapper);
+  FD_C_ClassifyResultWrapper* fd_c_classify_result_wrapper =
+      FD_C_CreateClassifyResultWrapper();
   auto& classify_result = CHECK_AND_CONVERT_FD_TYPE(
       ClassifyResultWrapper, fd_c_classify_result_wrapper);
-  return paddleclas_model->Predict(im, classify_result.get());
+
+  bool successful = paddleclas_model->Predict(im, classify_result.get());
+  if (successful) {
+    FD_C_ClassifyResult* res =
+        FD_C_ClassifyResultWrapperGetData(fd_c_classify_result_wrapper);
+    *fd_c_classify_result = *res;
+  }
+  return successful;
+}
+
+FD_C_Bool FD_C_PaddleClasModelWrapperInitialized(
+    FD_C_PaddleClasModelWrapper* fd_c_paddleclas_model_wrapper) {
+  auto& paddleclas_model = CHECK_AND_CONVERT_FD_TYPE(
+      PaddleClasModelWrapper, fd_c_paddleclas_model_wrapper);
+  return paddleclas_model->Initialized();
+}
+
+FD_C_ClassifyResult* FD_C_ClassifyResultToC(
+    fastdeploy::vision::ClassifyResult* classify_result) {
+  // Internal use, transfer fastdeploy::vision::ClassifyResult to
+  // FD_C_ClassifyResult
+  FD_C_ClassifyResult* fd_c_classify_result_data = new FD_C_ClassifyResult();
+  // copy label_ids
+  fd_c_classify_result_data->label_ids.size = classify_result->label_ids.size();
+  fd_c_classify_result_data->label_ids.data =
+      new int32_t[fd_c_classify_result_data->label_ids.size];
+  memcpy(fd_c_classify_result_data->label_ids.data,
+         classify_result->label_ids.data(),
+         sizeof(int32_t) * fd_c_classify_result_data->label_ids.size);
+  // copy scores
+  fd_c_classify_result_data->scores.size = classify_result->scores.size();
+  fd_c_classify_result_data->scores.data =
+      new float[fd_c_classify_result_data->scores.size];
+  memcpy(fd_c_classify_result_data->scores.data, classify_result->scores.data(),
+         sizeof(float) * fd_c_classify_result_data->scores.size);
+  fd_c_classify_result_data->type =
+      static_cast<FD_C_ResultType>(classify_result->type);
+  return fd_c_classify_result_data;
+}
+
+FD_C_Bool FD_C_PaddleClasModelWrapperBatchPredict(
+    FD_C_PaddleClasModelWrapper* fd_c_paddleclas_model_wrapper,
+    FD_C_OneDimMat imgs, FD_C_OneDimClassifyResult* results) {
+  std::vector<cv::Mat> imgs_vec;
+  std::vector<fastdeploy::vision::ClassifyResult> results_out;
+  for (int i = 0; i < imgs.size; i++) {
+    imgs_vec.push_back(*(reinterpret_cast<cv::Mat*>(imgs.data[i])));
+  }
+  auto& paddleclas_model = CHECK_AND_CONVERT_FD_TYPE(
+      PaddleClasModelWrapper, fd_c_paddleclas_model_wrapper);
+  bool successful = paddleclas_model->BatchPredict(imgs_vec, &results_out);
+  if (successful) {
+    // copy results back to FD_C_OneDimClassifyResult
+    results->size = results_out.size();
+    results->data = new FD_C_ClassifyResult[results->size];
+    for (int i = 0; i < results_out.size(); i++) {
+      results->data[i] = *FD_C_ClassifyResultToC(&results_out[i]);
+    }
+  }
+  return successful;
 }
 
 #ifdef __cplusplus
 }
-#endif
+#endif

c_api/fastdeploy_capi/vision/classification/ppcls/model.h

@@ -54,12 +54,36 @@ FASTDEPLOY_CAPI_EXPORT extern void FD_C_DestroyPaddleClasModelWrapper(
  *
  * \param[in] fd_c_paddleclas_model_wrapper pointer to FD_C_PaddleClasModelWrapper object
  * \param[in] img pointer to cv::Mat image
- * \param[in] fd_c_classify_result_wrapper pointer to FD_C_PaddleClasModelWrapper object, which stores the result.
+ * \param[in] fd_c_classify_result pointer to FD_C_ClassifyResult object, which stores the result.
  */
 
 FASTDEPLOY_CAPI_EXPORT extern FD_C_Bool FD_C_PaddleClasModelWrapperPredict(
-    __fd_take FD_C_PaddleClasModelWrapper* fd_c_paddleclas_model_wrapper,
-    FD_C_Mat img, FD_C_ClassifyResultWrapper* fd_c_classify_result_wrapper);
+    __fd_keep FD_C_PaddleClasModelWrapper* fd_c_paddleclas_model_wrapper,
+    FD_C_Mat img, FD_C_ClassifyResult* fd_c_classify_result_wrapper);
+
+/** \brief Check if the model is initialized successfully
+ *
+ * \param[in] fd_c_paddleclas_model_wrapper pointer to FD_C_PaddleClasModelWrapper object
+ *
+ * \return Return a bool of value true if initialized successfully
+ */
+
+FASTDEPLOY_CAPI_EXPORT extern FD_C_Bool FD_C_PaddleClasModelWrapperInitialized(
+    __fd_keep FD_C_PaddleClasModelWrapper* fd_c_paddleclas_model_wrapper);
+
+
+/** \brief Predict the classification results for a batch of input images
+   *
+   * \param[in] fd_c_paddleclas_model_wrapper pointer to FD_C_PaddleClasModelWrapper object
+   * \param[in] imgs The input image list, each element comes from cv::imread()
+   * \param[in] results The output classification result list
+   * \return true if the prediction successed, otherwise false
+   */
+FASTDEPLOY_CAPI_EXPORT extern FD_C_Bool FD_C_PaddleClasModelWrapperBatchPredict(
+            __fd_keep FD_C_PaddleClasModelWrapper* fd_c_paddleclas_model_wrapper,
+            FD_C_OneDimMat imgs,
+            FD_C_OneDimClassifyResult* results);
+
 
 #ifdef __cplusplus
 }  // extern "C"