[问答]

如何去实现sRK3399 Pro C++加载两个不同的模型呢

409

问答对人有帮助，内容完整，我也想知道答案 0 如何去实现sRK3399 Pro C++加载两个不同的模型呢？ 0
2022-2-11 06:17:08　　评论淘帖0 邀请回答您可以邀请以下用户，快速回答问题 × liutiefu 该类别下有 26 个回答。邀请回答 tigerwang711 该类别下有 20 个回答。邀请回答 ze55me 该类别下有 20 个回答。邀请回答杀狼000 该类别下有 19 个回答。邀请回答 zhuzb0754 该类别下有 19 个回答。邀请回答冰箱洗衣机该类别下有 18 个回答。邀请回答 YYXIAO 该类别下有 18 个回答。邀请回答 jjll652 该类别下有 17 个回答。邀请回答河神大人该类别下有 17 个回答。邀请回答 mede1001 该类别下有 17 个回答。邀请回答安德森大该类别下有 17 个回答。邀请回答 CDCNKA 该类别下有 17 个回答。邀请回答熊本熊该类别下有 16 个回答。邀请回答 YOYOOO 该类别下有 16 个回答。邀请回答麻酱该类别下有 16 个回答。邀请回答世态薄凉该类别下有 16 个回答。邀请回答 vinww特烦恼该类别下有 15 个回答。邀请回答 723662364d 该类别下有 15 个回答。邀请回答 musicalboy 该类别下有 15 个回答。邀请回答万航渡路该类别下有 15 个回答。邀请回答举报感谢相遇相关推荐 • 基于RK3399pro开发C++实现工业落地的步骤有哪些呢 1709 • 怎样去解决RK3399pro文件会进入不了Ubuntu系统的问题呢 2190 • RK3399pro使用python3读取USB摄像头数据出现错误怎么解决？ 1844 • 怎样去解决TB-RK3399pro成砖后Typec无法连接电脑的问题呢 1612 • RKNN格式模型怎样才能在RK3399Pro板子上成功运行呢 1487 • 求一种RK3399Pro AI平台解决方案 1279 • 为什么RK3399Pro平台YOLOv4 pytorch模型转RKNN模型会失败呢 1899 • 怎样去设计一款RK3399Pro嵌入式主机呢 1611 • 求一种RK3399Pro AI平台解决方案 1518 • 请问intel架构上转化的rknn模型可以在rk3399Pro上运行吗 8730 1个回答

答案对人有帮助，有参考价值 0 #include "rknn_api.h" #include #include #include #include #include "opencv2/core/core.hpp" #include "opencv2/imgproc/imgproc.hpp" #include "opencv2/highgui/highgui.hpp" using namespace std; void YOLO(string model_path) { FILE fp_ = fopen(model_path.c_str(), "rb"); if (fp_ == NULL) { printf("fopen %s fail!n", model_path.c_str()); return; } fseek(fp_, 0, SEEK_END); int model_len = ftell(fp_); void model_ = malloc(model_len); fseek(fp_, 0, SEEK_SET); if (model_len != fread(model_, 1, model_len, fp_)) { printf("fread %s fail!n", model_path.c_str()); free(model_); return; } rknn_context ctx = 0; int ret = rknn_init(&ctx, model_, model_len, RKNN_FLAG_PRIOR_MEDIUM); if (ret < 0) { printf("rknn_init fail! ret=%dn", ret); if (ctx) rknn_destroy(ctx); if (model_) free(model_); if (fp_) fclose(fp_); return; } // rknn_input_output_num io_num; ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); cv::Mat img = cv::imread("/home/toybrick/a.jpg"); int cols = img.cols, rows = img.rows; cv::resize(img, img, cv::Size(224, 224), (0, 0), (0, 0), cv::INTER_LINEAR); cv::cvtColor(img, img, cv::COLOR_BGR2RGB); // get result while (1) { rknn_input inputs[1]; memset(inputs, 0, sizeof(inputs)); inputs[0].index = 0; inputs[0].buf = img.data; inputs[0].size = 224 * 224 * 3; inputs[0].pass_through = false; inputs[0].type = RKNN_TENSOR_UINT8; inputs[0].fmt = RKNN_TENSOR_NHWC; ret = rknn_inputs_set(ctx, 1, inputs); if (ret < 0) { printf("rknn_input_set fail! ret=%dn", ret); if (ctx) rknn_destroy(ctx); if (model_) free(model_); if (fp_) fclose(fp_); return; } ret = rknn_run(ctx, nullptr); rknn_output outputs[io_num.n_output]; memset(outputs, 0, sizeof(outputs)); for (int i = 0; i < io_num.n_output; ++i) { outputs.want_float = true; outputs.is_prealloc = false; } ret = rknn_outputs_get(ctx, io_num.n_output, outputs, nullptr); if (ret < 0) { printf("rknn_outputs_get fail! ret=%dn", ret); if (ctx) rknn_destroy(ctx); if (model_) free(model_); if (fp_) fclose(fp_); return; } cout << outputs[0].size << endl; } } void ap(string model_path) { FILE fp_ = fopen(model_path.c_str(), "rb"); if (fp_ == NULL) { printf("fopen %s fail!n", model_path.c_str()); return; } fseek(fp_, 0, SEEK_END); int model_len = ftell(fp_); void model_ = malloc(model_len); fseek(fp_, 0, SEEK_SET); if (model_len != fread(model_, 1, model_len, fp_)) { printf("fread %s fail!n", model_path.c_str()); free(model_); return; } rknn_context ctx = 0; int ret = rknn_init(&ctx, model_, model_len, RKNN_FLAG_PRIOR_MEDIUM); if (ret < 0) { printf("rknn_init fail! ret=%dn", ret); if (ctx) rknn_destroy(ctx); if (model_) free(model_); if (fp_) fclose(fp_); return; } // rknn_input_output_num io_num; ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); uchar data[6406408]; // get result while (1) { rknn_input inputs[1]; memset(inputs, 0, sizeof(inputs)); inputs[0].index = 0; inputs[0].buf = data; inputs[0].size = 6406408; inputs[0].pass_through = false; inputs[0].type = RKNN_TENSOR_UINT8; inputs[0].fmt = RKNN_TENSOR_NHWC; ret = rknn_inputs_set(ctx, 1, inputs); if (ret < 0) { printf("rknn_input_set fail! ret=%dn", ret); if (ctx) rknn_destroy(ctx); if (model_) free(model_); if (fp_) fclose(fp_); return; } ret = rknn_run(ctx, nullptr); rknn_output outputs[io_num.n_output]; memset(outputs, 0, sizeof(outputs)); for (int i = 0; i < io_num.n_output; ++i) { outputs.want_float = true; outputs.is_prealloc = false; } ret = rknn_outputs_get(ctx, io_num.n_output, outputs, nullptr); if (ret < 0) { printf("rknn_outputs_get fail! ret=%dn", ret); if (ctx) rknn_destroy(ctx); if (model_) free(model_); if (fp_) fclose(fp_); return; } cout << outputs[0].size << endl; } } int main() { string model_path("/home/toybrick/coco.rknn"); thread t(YOLO, model_path); string modelpath("/home/toybrick/d.rknn"); thread t1(ap, modelpath); t.join(); t1.join(); return 0; } CMakeLists.txt cmake_minimum_required(VERSION 3.10) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11") find_package(OpenCV REQUIRED) include_directories( ${OpenCV_INCLUDE_DIRS} include ) add_executable(duo src/duo.cc) target_link_libraries(duo ${PROJECT_SOURCE_DIR}/librknn_api.so ${OpenCV_LIBRARIES} pthread )

2022-2-11 14:18:23 评论举报黄淳