《meanshift 代码》由会员分享，可在线阅读，更多相关《meanshift 代码（6页珍藏版）》请在金锄头文库上搜索。
1、% Adam Kukucka % Zach Clay % Marcelo Molina % CSE 486 Project 3 function trackmov probmov centers = camshift % * % initialize vari ables % * rmin = 0; %min row value for search window rmax = 0; %max row value for search window cmin = 0; %min col value for search window cmax = 0; %max col value for s
2、earch window numofframes = 0; %number of frames in the avi threshold = 1; %threshold for convergence centerold = 0 0; %for convergence. previous center of window centernew = 0 0; %for convergence. new center of window % * % Pre code. load movie and select initial frame % * % prompt user for avi file
3、 name user_entry = input(Please enter an avi filename: ,s); % load the avi file. handle is M M = aviread(user_entry); % get number of frames dontneed numberofframes = size(M); % initialize matrix to hold center coordinates imagecenters = zeros(numberofframes, 2); % extract the first frame from the a
4、vi Frame1 = M(1,1); Image1 = frame2im(Frame1); % * images(:, :, numberofframes) = G(:,:); % get search window for first frame cmin, cmax, rmin, rmax = select( Image1 ); cmin = round(cmin); cmax = round(cmax); rmin = round(rmin); rmax = round(rmax); wsize(1) = abs(rmax - rmin); wsize(2) = abs(cmax -
5、cmin); % create histogram % translate to hsv hsvimage = rgb2hsv(Image1); % pull out the h huenorm = hsvimage(:,:,1);% scale to 0 to 255 hue = huenorm*255; % set unit type hue=uint8(hue); % Getting Histogram of Image: histogram = zeros(256); for i=rmin:rmax for j=cmin:cmax index = uint8(hue(i,j)+1);
6、%count number of each pixel histogram(index) = histogram(index) + 1; end end % * % Algorithm from pdf % * aviobj1 = avifile(example3.avi); aviobj2 = avifile(example4.avi); % for each frame for i = 1:200 disp(Processing frame); disp(i); Frame = M(1, i); I = frame2im(Frame);% translate to hsv hsvimage
7、 = rgb2hsv(I); % pull out the h huenorm = hsvimage(:,:,1); % scale to 0 to 255 hue = huenorm*255; % set unit type hue=uint8(hue);rows cols = size(hue);% choose initial search window % the search window is (cmin, rmin) to (cmax, rmax)% create a probability map probmap = zeros(rows, cols);for r=1:rows
8、 for c=1:cols if(hue(r,c) = 0) probmap(r,c)= histogram(hue(r,c); end end end probmap = probmap/max(max(probmap); probmap = probmap*255;count = 0;rowcenter = 0; % any number just so it runs through at least twice colcenter = 0; rowcenterold = 30; colcenterold = 30; % Mean Shift for 15 iterations or u
9、ntil convergence(the center doesnt % change) while (abs(rowcenter - rowcenterold) 2) % disp(j); rmin = rmin - 7; %increase window size and check for center rmax = rmax + 7; cmin = cmin - 7; cmax = cmax + 7;rowcenterold = rowcenter; %save old center for convergence check colcenterold = colcenter; row
10、center colcenter M00 = meanshift(I, rmin, rmax, cmin,. cmax, probmap); % given image (I), search window(rmin rmax cmin cmax) % returns new center (colcenter, rowcenter) for window and % zeroth moment (Moo)% redetermine window around new center rmin = round(rowcenter - wsize(1)/2); rmax = round(rowce
11、nter + wsize(1)/2); cmin = round(colcenter - wsize(2)/2); cmax = round(colcenter + wsize(2)/2); wsize(1) = abs(rmax - rmin); wsize(2) = abs(cmax - cmin);count = count + 1; end% mark center on image %save image G = .2989*I(:,:,1). +.5870*I(:,:,2). +.1140*I(:,:,3); trackim=G;%make box of current searc
12、h window on saved image for r= rmin:rmax trackim(r, cmin) = 255; trackim(r, cmax) = 255; end for c= cmin:cmax trackim(rmin, c) = 255; trackim(rmax, c) = 255; end aviobj1 = addframe(aviobj1,trackim); aviobj2 = addframe(aviobj2,probmap);%create image movie, and probability map movie trackmov(:,:,i)= t
13、rackim(:,:); probmov(:,:,i) = probmap(:,:);% save center coordinates as an x, y by doing col, row centers(i,:) = colcenter rowcenter; % Set window size = 2 * (Moo/256)1/2 windowsize = 2 * (M00/256).5;% get side length . window size is an area so sqrt(Area)=sidelength sidelength = sqrt(windowsize);%
14、determine rmin, rmax, cmin, cmax rmin = round(rowcenter-sidelength/2); rmax = round(rowcenter+sidelength/2); cmin = round(colcenter-sidelength/2); cmax = round(colcenter+sidelength/2); wsize(1) = abs(rmax - rmin); wsize(2) = abs(cmax - cmin); end% end for loop% Adam Kukucka % Zach Clay % Marcelo Molina % CSE 486 Project 3 function rowcenter colcenter M00 = meanshift(I, rmin, rmax, cmin,. cmax, probmap) %inputs % rmin, rmax, cmin, cmax are the coordiantes of the window % I is the image %outputs % colcenter rowcenter are the new center coordinates % Moo is the zeroth mean % *
