主要内容

本页采用了机器翻译。点击此处可查看英文原文。

实例分割

使用 SOLOv2、Mask R-CNN 和 SAM 等预训练 AI 模型标注真实值并执行实例分割,或使用迁移学习训练自定义模型

Computer Vision Toolbox™ 中的实例分割工具支持您检测、分类和分割图像中的单个目标,即使多个目标相互重叠也可以。您可以首先使用图像标注器视频标注器创建标注真实值,它们支持使用多边形或矩形 ROI 对目标实例进行交互式和 AI 辅助注解。有关详细信息,请参阅Label Objects Using Polygons for Instance Segmentation

工具箱提供了预训练实例分割网络,如 SOLOv2 和 Mask R-CNN。您可以直接使用这些模型进行推断,或通过迁移学习将其适配到特定应用。有关详细信息,请参阅Get Started with Instance Segmentation Using Deep LearningGet Started with SOLOv2 for Instance Segmentation。对于类别无关的实例分割,工具箱通过 imsegsam 函数和 segmentAnythingModel 目标支持分割一切模型 (SAM)。

为了准备训练数据,工具箱提供了用于管理和组织数据集以及数据增强和预处理的实用工具。有关详细信息,请参阅Postprocess Exported Labels for Instance Segmentation Training

使用预训练或自定义模型生成预测结果后,您可以评估实例分割性能,并生成有关分割准确性、目标级精确率以及不同目标大小下性能表现的详细分析结果。这些度量有助于评估掩膜预测和边界框定位的质量。有关详细信息,请参阅 evaluateInstanceSegmentation

工具箱还支持利用 Pose Mask R-CNN 框架,通过实例分割实现三维目标位姿估计,从而能够对目标的朝向和结构进行细粒度分析。有关详细信息,请参阅Perform 6-DoF Pose Estimation for Bin Picking Using Deep Learning

Instance segmentation using SOLOv2: Left — A segmented and labeled road scenario using a sample modified RGB image from the CamVid data set, Right — A segmented image of PVC pipe connectors

App

图像标注器为计算机视觉应用标注图像
视频标注器Label video for computer vision applications

函数

全部展开

SOLOv2

solov2Segment objects using SOLOv2 instance segmentation network (自 R2023b 起)
segmentObjectsSegment objects using SOLOv2 instance segmentation (自 R2023b 起)

Mask R-CNN

maskrcnnDetect objects using Mask R-CNN instance segmentation (自 R2021b 起)
segmentObjectsSegment objects using Mask R-CNN instance segmentation (自 R2021b 起)

分割一切模型 (SAM)

imsegsamPerform automatic full image segmentation using Segment Anything Model 2 (SAM 2) (自 R2024b 起)
segmentAnythingModelPretrained Segment Anything Model 2 (SAM 2) for image segmentation (自 R2024a 起)

加载训练数据

boxLabelDatastoreDatastore for bounding box label data
groundTruthGround truth label data
imageDatastore图像数据的数据存储
combine合并来自多个数据存储的数据

训练实例分割网络

trainSOLOV2Train SOLOv2 network to perform instance segmentation (自 R2023b 起)
trainMaskRCNNTrain Mask R-CNN network to perform instance segmentation (自 R2022a 起)

增强和预处理训练数据

poly2mask将感兴趣的区域 (ROI) 多边形转换为区域掩膜
bwboundaries跟踪二值图像中的目标边界
balanceBoxLabelsBalance bounding box labels for object detection
bboxcropCrop bounding boxes
bboxeraseRemove bounding boxes
bboxresizeResize bounding boxes
bboxwarp对边界框应用几何变换
bbox2pointsConvert rectangle to corner points list
imwarp对图像应用几何变换
imcrop裁剪图像
imresize调整图像大小
randomAffine2dCreate randomized 2-D affine transformation
centerCropWindow2dCreate rectangular center cropping window
randomWindow2dRandomly select rectangular region in image
evaluateInstanceSegmentationEvaluate instance segmentation data set against ground truth (自 R2022b 起)
instanceSegmentationMetricsInstance segmentation quality metrics (自 R2022b 起)
metricsByAreaEvaluate instance segmentation across object mask size ranges (自 R2023b 起)
insertObjectMask Insert masks in image or video stream
insertObjectAnnotationAnnotate truecolor or grayscale image or video
insertShape在图像或视频中插入图形
insertText在图像或视频中插入文字
showShapeDisplay shapes on image, video, or point cloud
posemaskrcnnPredict object pose using Pose Mask R-CNN pose estimation (自 R2024a 起)
predictPoseEstimate object pose using Pose Mask R-CNN deep learning network (自 R2024a 起)
trainPoseMaskRCNNTrain Pose Mask R-CNN network to perform pose estimation (自 R2024a 起)

主题

快速入门

为实例分割创建真实值

为实例分割准备训练数据

精选示例