B站视频讲解https://www.bilibili.com/video/BV1zNvkeAEZy?p=23
GitHub源码https://github.com/KeepTryingTo/QT_Learning/tree/main/QtOpenCV/QtOpenCV_Classify
本文主要是利用cv::dnn::ClassificationModel实现图像分类,和QT 6.6.0 基于OpenCV加载.pth模型文件转换之后的ONNX模型文件,并且实现图像分类 没有本质的区别,但是实现的方式有所区别,建议读者先去看这篇博文之后再来看本文。关于ONNX模型文件的转换建议看PyTorch 训练之后的网络模型.pth转.onnx文件并对图像进行预测。
第一步: 模型加载
第二步:图像预处理
第三步:预测类别概率以及对应类别索引
附加内容
QT 6.6.0基于OpenCVONNX模型文件实现图像分类