MATLAB UNET代码

以下是一个简单的 MATLAB UNET 代码，用于图像语义分割任务：

% 加载训练数据和标签
data = imageDatastore('path/to/training/images');
label = pixelLabelDatastore('path/to/training/labels');

% 创建并训练 UNET 模型
numClasses = 2; % 二分类任务
inputSize = [256 256 3]; % 输入图像大小
layers = unetLayers(inputSize, numClasses);
options = trainingOptions('adam', 'InitialLearnRate', 1e-3, 'MaxEpochs', 20);
net = trainNetwork(data, label, layers, options);

% 加载测试图像
testImage = imread('path/to/test/image');

% 对测试图像进行分割
predictedLabels = semanticseg(testImage, net);

% 显示分割结果
imshow(testImage);
hold on;
mask = predictedLabels == 'object'; % 选择目标类别
visboundaries(mask, 'Color', 'r'); % 显示目标边界

请注意，此代码仅用于演示目的，您需要替换`path/to/training/images` 和 `path/to/training/labels` 为您自己的训练数据路径。