4 种方法快速区分:分类 (种类预测) / 目标检测 / 分割 / 回归

📅 2026/7/30 4:12:46
4 种方法快速区分:分类 (种类预测) / 目标检测 / 分割 / 回归
一、4 种方法快速区分:分类 (种类预测) / 目标检测 / 分割 / 回归方法 1:看模型输出张量 shape(最准、通用,pt/pb/caffemodel/onnx 全部适用)1)图像分类(种类预测)作用:整张图片只输出 1 组类别概率,没有坐标框 输出维度两种常见格式:[1, num_classes]1:batch 一张图num_classes:类别总数(比如猫狗二分类 = 2,ImageNet 千分类 = 1000)[1, num_classes, 1, 1](早期 Caffe 分类模型格式) 特征:无 xy 坐标、无框宽高,只有每一类的置信度。 示例:输出(1,1000)→ 千种类别预测。2)目标检测(YOLO/SSD/FasterRCNN)输出一定包含框坐标 + 置信度 + 类别,shape 格式: SSD/Caffe 检测输出:[1,1,N,7]N = 检测框数量,最后一维 7 个值:[0, class_id, conf, x1, y1, x2, y2]YOLO 输出:[1, num_anchors, num_classes+5]末尾 5:x,y,w,h,conf 特征:存在坐标、框尺寸,代表定位 + 分类。3)语义分割 / 实例分割输出 shape:[1, num_classes, H, W]H、W 和原图分辨率一致,每个像素输出类别; 可视化是彩色掩码图,无检测框。4)回归模型(预测数值,不是种类)输出[1, n],n 是预测数值数量(角度、距离、年龄、关键点坐标),没有类别概率。方法 2:看网络文件结构(