天天看點

matlab在圖檔上畫框

之前寫過一個MATLAB在圖檔上畫框的代碼, http://blog.csdn.net/carson2005/article/details/17262811

最近使用後發現效果不好,貌似有個bug,這裡做更改,

function [state result] = draw_rect(img, rectVec, showOrNot)

%img: input color image
%rectVec: input vector of rect draw on image. element in rectVec: x, y,
%width, height


rgb = [255 0 0];
result = img;

[imgH, imgW] = size(img);

if size(img,3) == 3 %draw on color image
    for k=1:3
        for i=1:size(rectVec,1)
            if( rectVec(i,1)>=0 && rectVec(i,2)>=0 && rectVec(i,1)+rectVec(i,3)<=imgW && rectVec(i,2)+rectVec(i,4)<=imgH )
                result( rectVec(i,2),rectVec(i,1):(rectVec(i,1)+rectVec(i,3)),k ) = rgb(1,k);%top line
                result( rectVec(i,2)+1,rectVec(i,1):(rectVec(i,1)+rectVec(i,3)),k ) = rgb(1,k);
                
                result( rectVec(i,2)+rectVec(i,4),rectVec(i,1):(rectVec(i,1)+rectVec(i,3)),k ) = rgb(1,k);%bottom line
                result( rectVec(i,2)+rectVec(i,4)-1,rectVec(i,1):(rectVec(i,1)+rectVec(i,3)),k ) = rgb(1,k);
                
                result( rectVec(i,2):rectVec(i,2)+rectVec(i,4), rectVec(i,1),k ) = rgb(1,k);%left line
                result( rectVec(i,2):rectVec(i,2)+rectVec(i,4), rectVec(i,1)+1,k ) = rgb(1,k);
                
                result( rectVec(i,2):rectVec(i,2)+rectVec(i,4), rectVec(i,1)+rectVec(i,3),k ) = rgb(1,k);%right line
                result( rectVec(i,2):rectVec(i,2)+rectVec(i,4), rectVec(i,1)+rectVec(i,3)-1,k ) = rgb(1,k);
                
            end
        end
    end
end

state = 1;

if showOrNot == 1
    figure;
    imshow(result);
end