python调用caffe模型_Toybrick-开源社区-人工智能-调用python脚本转caffe模型为rknn 模型...

想实现一个图片数据的输入,然后输出一个特征结结果,一下写法对吗?

int DealWithModel(cv::Mat img, vector& V1)

{

const int img_width = 96;                                                  /*图像尺寸*/

const int img_height = 112;

const int img_channels = 3;                                                /*图像量化通道*/

const int input_index = 0;                                                 // node name "Preprocessor/sub"

if (img.cols != img_width || img.rows != img_height)                       /*如果图像的大小和定义的不一样,则进行重新设置大小*/

{

cv::resize(img, img, cv::Size(img_width, img_height), (0, 0), (0, 0), cv::INTER_LINEAR);

}

FILE* fp = fopen(model_path, "rb");                                        /*加载模型*/

if (fp == NULL)

{

printf("fopen %s fail!\n", model_path);

return -1;

}

fseek(fp, 0, SEEK_END);                                                    /*标记fp从文件头开始到文件末尾*/

int model_len = ftell(fp);                                                 /*获取fp从头到尾的文件长度*/

void* model = malloc(model_len);                                           /*动态申请内存*/

fseek(fp, 0, SEEK_SET);                                                    /*把文件指针指向文件的开始*/

if (model_len != fread(model

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值