尧图网站设计 尧图网站设计YAOTU DESIGN
ARTICLE DETAIL

资讯详情

深耕网站设计与一线实操的经验洞察。

OpenCV C++基于CNN模型的图像分类

OpenCV C++基于CNN模型的图像分类 这个程序使用OpenCV加载一个Caffe模型进行图像分类。输入图像左边是一只小猫咪右边是个航天飞机程序运行后输入如下id为285的是埃及猫id为812的是航天飞机代码int main() { Mat img imread(cat.jpg); Net net readNetFromCaffe(SqueezeNet_deploy.prototxt, SqueezeNet.caffemodel); Mat input_blob blobFromImage( img, 1.0, Size(416, 416), Scalar(104, 117, 123), false); Mat prob; net.setInput(input_blob); prob net.forward(predictions); int prob_size[3] {1000, 1, 1}; Mat prob_data(3, prob_size, CV_32F, prob.ptrfloat(0)); double max_prob -1.0; int class_idx -1; for (int idx 0; idx prob.size[1]; idx) { double current_prob prob_data.atfloat(idx, 0, 0); if (current_prob max_prob) { max_prob current_prob; class_idx idx; } } cout Best class Index: class_idx \n; cout Probability: max_prob * 100.0 \n; }
返回列表