matlab drawrect,drawRect.m · gavine/SSD-Single-Shot-Detector-in-Matlab - Gitee.com

% 此函数非本人编写

% 参考自：https://www.cnblogs.com/GarfieldEr007/p/5598861.html

function [ dest ] = drawRect( src, pt, wSize, lineSize, color )

%简介：

% %将图像画上有颜色的框图，如果输入是灰度图，先转换为彩色图像，再画框图

% 图像矩阵

% 行向量方向 是 y

% 列向量方向 是 x

%----------------------------------------------------------------------

%输入：

% src： 原始图像，可以为灰度图，可为彩色图

% pt： 左上角坐标 [x1, y1]

% wSize： 框的大小 [wx, wy]

% lineSize： 线的宽度

% color： 线的颜色 [r, g, b]

%----------------------------------------------------------------------

%输出：

% dest： 画好了的图像

%----------------------------------------------------------------------

%flag=1: 有缺口的框

%flag=2: 无缺口的框

flag = 1;

%判断输入参数个数

if nargin < 5

color = [255 255 0];

end

if nargin < 4

lineSize = 1;

end

if nargin < 3

disp('输入参数不够 !!!');

return;

end

%判断框的边界问题

[yA, xA, z] = size(src);

x1 = pt(1);

y1 = pt(2);

wx = wSize(1);

wy = wSize(2);

if x1>xA || ...

y1>yA||...

(x1+wx)>xA||...

(y1+wy)>yA

disp('画的框将超过图像 !!!');

return;

end

%如果是单通道的灰度图，转成3通道的图像

if 1==z

dest(:, : ,1) = src;

dest(:, : ,2) = src;

dest(:, : ,3) = src;

else

dest = src;

end

%开始画框图

for c = 1 : 3 %3个通道，r，g，b分别画

for dl = 1 : lineSize %线的宽度，线条是向外面扩展的

d = dl - 1;

if 1==flag %有缺口的框

try

dest( y1-d , x1:(x1+wx) , c ) = color(c); %上方线条

dest( y1+wy+d , x1:(x1+wx) , c ) = color(c); %下方线条

dest( y1:(y1+wy) , x1-d , c ) = color(c); %左方线条

dest( y1:(y1+wy) , x1+wx+d , c ) = color(c); %左方线条

catch

continue

end

elseif 2==flag %无缺口的框

dest( y1-d , (x1-d):(x1+wx+d) , c ) = color(c); %上方线条

dest( y1+wy+d , (x1-d):(x1+wx+d) , c ) = color(c); %下方线条

dest( (y1-d):(y1+wy+d) , x1-d , c ) = color(c); %左方线条

dest( (y1-d):(y1+wy+d) , x1+wx+d , c ) = color(c); %左方线条

end

end

end %主循环尾

end %函数尾

一键复制

编辑

Web IDE

原始数据

按行查看

历史