Skip to content

Commit

Permalink
format cpp and cuda code
Browse files Browse the repository at this point in the history
  • Loading branch information
@huzhenhong
huzhenhong committed Dec 11, 2023
1 parent aeac6b3 commit 29aa2d2
Show file tree
Hide file tree
Showing 559 changed files with 72,407 additions and 59,912 deletions.
403 changes: 251 additions & 152 deletions .clang-format
View file

Large diffs are not rendered by default.

196 changes: 105 additions & 91 deletions csrc/mmdeploy/apis/c/mmdeploy/classifier.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,118 +16,132 @@
using namespace mmdeploy;
using namespace std;

int mmdeploy_classifier_create(mmdeploy_model_t model, const char* device_name, int device_id,
mmdeploy_classifier_t* classifier) {
mmdeploy_context_t context{};
auto ec = mmdeploy_context_create_by_device(device_name, device_id, &context);
if (ec != MMDEPLOY_SUCCESS) {
int mmdeploy_classifier_create(mmdeploy_model_t model, const char* device_name, int device_id, mmdeploy_classifier_t* classifier)
{
mmdeploy_context_t context{};
auto ec = mmdeploy_context_create_by_device(device_name, device_id, &context);
if (ec != MMDEPLOY_SUCCESS)
{
return ec;
}
ec = mmdeploy_classifier_create_v2(model, context, classifier);
mmdeploy_context_destroy(context);
return ec;
}
ec = mmdeploy_classifier_create_v2(model, context, classifier);
mmdeploy_context_destroy(context);
return ec;
}

int mmdeploy_classifier_create_by_path(const char* model_path, const char* device_name,
int device_id, mmdeploy_classifier_t* classifier) {
mmdeploy_model_t model{};
int mmdeploy_classifier_create_by_path(const char* model_path, const char* device_name, int device_id, mmdeploy_classifier_t* classifier)
{
mmdeploy_model_t model{};

if (auto ec = mmdeploy_model_create_by_path(model_path, &model)) {
if (auto ec = mmdeploy_model_create_by_path(model_path, &model))
{
return ec;
}
auto ec = mmdeploy_classifier_create(model, device_name, device_id, classifier);
mmdeploy_model_destroy(model);
return ec;
}
auto ec = mmdeploy_classifier_create(model, device_name, device_id, classifier);
mmdeploy_model_destroy(model);
return ec;
}

int mmdeploy_classifier_create_v2(mmdeploy_model_t model, mmdeploy_context_t context,
mmdeploy_classifier_t* classifier) {
return mmdeploy_pipeline_create_from_model(model, context, (mmdeploy_pipeline_t*)classifier);
int mmdeploy_classifier_create_v2(mmdeploy_model_t model, mmdeploy_context_t context, mmdeploy_classifier_t* classifier)
{
return mmdeploy_pipeline_create_from_model(model, context, (mmdeploy_pipeline_t*)classifier);
}

int mmdeploy_classifier_create_input(const mmdeploy_mat_t* mats, int mat_count,
mmdeploy_value_t* value) {
return mmdeploy_common_create_input(mats, mat_count, value);
int mmdeploy_classifier_create_input(const mmdeploy_mat_t* mats, int mat_count, mmdeploy_value_t* value)
{
return mmdeploy_common_create_input(mats, mat_count, value);
}

int mmdeploy_classifier_apply(mmdeploy_classifier_t classifier, const mmdeploy_mat_t* mats,
int mat_count, mmdeploy_classification_t** results,
int** result_count) {
wrapped<mmdeploy_value_t> input;
if (auto ec = mmdeploy_classifier_create_input(mats, mat_count, input.ptr())) {
return ec;
}
wrapped<mmdeploy_value_t> output;
if (auto ec = mmdeploy_classifier_apply_v2(classifier, input, output.ptr())) {
return ec;
}
if (auto ec = mmdeploy_classifier_get_result(output, results, result_count)) {
return ec;
}
return MMDEPLOY_SUCCESS;
int mmdeploy_classifier_apply(mmdeploy_classifier_t classifier, const mmdeploy_mat_t* mats, int mat_count, mmdeploy_classification_t** results, int** result_count)
{
wrapped<mmdeploy_value_t> input;
if (auto ec = mmdeploy_classifier_create_input(mats, mat_count, input.ptr()))
{
return ec;
}
wrapped<mmdeploy_value_t> output;
if (auto ec = mmdeploy_classifier_apply_v2(classifier, input, output.ptr()))
{
return ec;
}
if (auto ec = mmdeploy_classifier_get_result(output, results, result_count))
{
return ec;
}
return MMDEPLOY_SUCCESS;
}

int mmdeploy_classifier_apply_v2(mmdeploy_classifier_t classifier, mmdeploy_value_t input,
mmdeploy_value_t* output) {
return mmdeploy_pipeline_apply((mmdeploy_pipeline_t)classifier, input, output);
int mmdeploy_classifier_apply_v2(mmdeploy_classifier_t classifier, mmdeploy_value_t input, mmdeploy_value_t* output)
{
return mmdeploy_pipeline_apply((mmdeploy_pipeline_t)classifier, input, output);
}

int mmdeploy_classifier_apply_async(mmdeploy_classifier_t classifier, mmdeploy_sender_t input,
mmdeploy_sender_t* output) {
return mmdeploy_pipeline_apply_async((mmdeploy_pipeline_t)classifier, input, output);
int mmdeploy_classifier_apply_async(mmdeploy_classifier_t classifier, mmdeploy_sender_t input, mmdeploy_sender_t* output)
{
return mmdeploy_pipeline_apply_async((mmdeploy_pipeline_t)classifier, input, output);
}

int mmdeploy_classifier_get_result(mmdeploy_value_t output, mmdeploy_classification_t** results,
int** result_count) {
if (!output || !results || !result_count) {
return MMDEPLOY_E_INVALID_ARG;
}
try {
Value& value = Cast(output)->front();

auto classify_outputs = from_value<vector<mmcls::Labels>>(value);

vector<int> _result_count;
_result_count.reserve(classify_outputs.size());

for (const auto& cls_output : classify_outputs) {
_result_count.push_back((int)cls_output.size());
int mmdeploy_classifier_get_result(mmdeploy_value_t output, mmdeploy_classification_t** results, int** result_count)
{
if (!output || !results || !result_count)
{
return MMDEPLOY_E_INVALID_ARG;
}

auto total = std::accumulate(begin(_result_count), end(_result_count), 0);

std::unique_ptr<int[]> result_count_data(new int[_result_count.size()]{});
std::copy(_result_count.begin(), _result_count.end(), result_count_data.get());

std::unique_ptr<mmdeploy_classification_t[]> result_data(
new mmdeploy_classification_t[total]{});
auto result_ptr = result_data.get();
for (const auto& cls_output : classify_outputs) {
for (const auto& label : cls_output) {
result_ptr->label_id = label.label_id;
result_ptr->score = label.score;
++result_ptr;
}
try
{
Value& value = Cast(output)->front();

auto classify_outputs = from_value<vector<mmcls::Labels>>(value);

vector<int> _result_count;
_result_count.reserve(classify_outputs.size());

for (const auto& cls_output : classify_outputs)
{
_result_count.push_back((int)cls_output.size());
}

auto total = std::accumulate(begin(_result_count), end(_result_count), 0);

std::unique_ptr<int[]> result_count_data(new int[_result_count.size()]{});
std::copy(_result_count.begin(), _result_count.end(), result_count_data.get());

std::unique_ptr<mmdeploy_classification_t[]> result_data(
new mmdeploy_classification_t[total]{});
auto result_ptr = result_data.get();
for (const auto& cls_output : classify_outputs)
{
for (const auto& label : cls_output)
{
result_ptr->label_id = label.label_id;
result_ptr->score = label.score;
++result_ptr;
}
}

*result_count = result_count_data.release();
*results = result_data.release();

return MMDEPLOY_SUCCESS;
}

*result_count = result_count_data.release();
*results = result_data.release();

return MMDEPLOY_SUCCESS;
} catch (const std::exception& e) {
MMDEPLOY_ERROR("unhandled exception: {}", e.what());
} catch (...) {
MMDEPLOY_ERROR("unknown exception caught");
}
return MMDEPLOY_E_FAIL;
catch (const std::exception& e)
{
MMDEPLOY_ERROR("unhandled exception: {}", e.what());
}
catch (...)
{
MMDEPLOY_ERROR("unknown exception caught");
}
return MMDEPLOY_E_FAIL;
}

void mmdeploy_classifier_release_result(mmdeploy_classification_t* results, const int* result_count,
int count) {
delete[] results;
delete[] result_count;
void mmdeploy_classifier_release_result(mmdeploy_classification_t* results, const int* result_count, int count)
{
delete[] results;
delete[] result_count;
}

void mmdeploy_classifier_destroy(mmdeploy_classifier_t classifier) {
mmdeploy_pipeline_destroy((mmdeploy_pipeline_t)classifier);
void mmdeploy_classifier_destroy(mmdeploy_classifier_t classifier)
{
mmdeploy_pipeline_destroy((mmdeploy_pipeline_t)classifier);
}
Loading

0 comments on commit 29aa2d2

Please sign in to comment.