[max_resp_row, max_row] = max(response, [], 1); [init_max_response, max_col] = max(max_resp_row, [], 2); max_row_perm = permute(max_row, [2 3 1]); col = max_col(:)'; row = max_row_perm(sub2ind(size(max_row_perm), col, 1:size(response,3))); trans_row = mod(row - 1 + floor((use_sz(1)-1)/2), use_sz(1)) - floor((use_sz(1)-1)/2); trans_col = mod(col - 1 + floor((use_sz(2)-1)/2), use_sz(2)) - floor((use_sz(2)-1)/2); init_pos_y = permute(2pi * trans_row / use_sz(1), [1 3 2]); init_pos_x = permute(2pi * trans_col / use_sz(2), [1 3 2]); max_pos_y = init_pos_y; max_pos_x = init_pos_x; % pre-compute complex exponential exp_iky = exp(bsxfun(@times, 1i * ky, max_pos_y)); exp_ikx = exp(bsxfun(@times, 1i * kx, max_pos_x)); % gradient_step_size = gradient_step_size / prod(use_sz); ky2 = ky.ky; kx2 = kx.kx; iter = 1; while iter <= iterations % Compute gradient ky_exp_ky = bsxfun(@times, ky, exp_iky); kx_exp_kx = bsxfun(@times, kx, exp_ikx); y_resp = mtimesx(exp_iky, responsef, 'speed'); resp_x = mtimesx(responsef, exp_ikx, 'speed'); grad_y = -imag(mtimesx(ky_exp_ky, resp_x, 'speed')); grad_x = -imag(mtimesx(y_resp, kx_exp_kx, 'speed')); ival = 1i * mtimesx(exp_iky, resp_x, 'speed'); H_yy = real(-mtimesx(bsxfun(@times, ky2, exp_iky), resp_x, 'speed') + ival); H_xx = real(-mtimesx(y_resp, bsxfun(@times, kx2, exp_ikx), 'speed') + ival); H_xy = real(-mtimesx(ky_exp_ky, mtimesx(responsef, kx_exp_kx, 'speed'), 'speed')); det_H = H_yy . H_xx - H_xy . H_xy; % Compute new position using newtons method max_pos_y = max_pos_y - (H_xx .* grad_y - H_xy .* grad_x) ./ det_H; max_pos_x = max_pos_x - (H_yy .* grad_x - H_xy .* grad_y) ./ det_H; % Evaluate maximum exp_iky = exp(bsxfun(@times, 1i * ky, max_pos_y)); exp_ikx = exp(bsxfun(@times, 1i * kx, max_pos_x)); iter = iter + 1; end max_response = 1 / prod(use_sz) * real(mtimesx(mtimesx(exp_iky, responsef, 'speed'), exp_ikx, 'speed')); % check for scales that have not increased in score ind = max_response < init_max_response; max_response(ind) = init_max_response(ind); max_pos_y(ind) = init_pos_y(ind); max_pos_x(ind) = init_pos_x(ind); [max_scale_response, sind] = max(max_response(:)); disp_row = (mod(max_pos_y(1,1,sind) + pi, 2pi) - pi) / (2pi) * use_sz(1); disp_col = (mod(max_pos_x(1,1,sind) + pi, 2pi) - pi) / (2pi) * use_sz(2); end代码详解
4869
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
