写出SAM分割算法的matlab代码

以下是SAM分割算法的matlab代码实现：

function [seg_map] = sam_seg(img, ref, threshold)
% SAM分割算法
% img：待分割的图像
% ref：参考光谱
% threshold：阈值
% seg_map：分割后的图像

% 将图像转化为double类型
img = im2double(img);
ref = im2double(ref);

% 图像尺寸
[h, w, ~] = size(img);

% 计算每个像素的光谱角度
img_vec = reshape(img, [], 3);
ref_vec = reshape(ref, [], 3);
img_norm = sqrt(sum(img_vec.^2, 2));
ref_norm = sqrt(sum(ref_vec.^2, 2));
cos_theta = sum(img_vec.*ref_vec, 2)./(img_norm.*ref_norm);
theta = acosd(cos_theta);

% 分割
seg_map = zeros(h, w);
seg_map(theta<=threshold) = 1;
seg_map(theta>threshold) = 2;

% 调整分割结果
seg_map = reshape(seg_map, h, w);
seg_map = imfill(seg_map, 'holes');
seg_map = imopen(seg_map, strel('disk', 5));

% 可视化
figure;
imshow(seg_map, []);
colormap(jet(2));
colorbar;
end

其中，`img`为待分割的图像，`ref`为参考光谱，`threshold`为阈值。函数返回分割后的图像`seg_map`。