скрипт для расчета границ между элементами матрицы

Следующий скрипт дает мне сумму границ каждого элемента матрицы. границы рассчитываются относительно элементов со значением 1, которые находятся рядом друг с другом. произведение этого суммирования называется контактным периметром.

но есть ли другой способ, которым я могу обобщить или векторизовать мой необработанный и простой сценарий? Я задаю этот запрос, потому что моя реальная матрица очень велика, а использование for увеличивает время расчета.

Спасибо.

a1=[1 1 0 1 0 1;
    0 1 1 0 0 1;
    1 1 0 1 0 1;
    1 1 0 0 1 0;
    0 0 0 1 1 1]
m=5
n=6
cmp=zeros(m,n)
cmp1=zeros(m,n)
for i=1:m-1
    for j=1:n
        if a1(i,j)==a1(i+1,j) && a1(i,j)==1
            cmp(i,j)=1
        end
    end
    for i=1:m
        for j=1:n-1
            if a1(i,j)==a1(i,j+1) && a1(i,j)==1
                cmp1(i,j)=1
            end
        end
    end
end
cmtotal=cmp+cmp1
pc=sum(sum(cmtotal))

matlab performance matrix vectorization
person user3760558    schedule 21.01.2015    source источник

Ответы (1)


arrow_upward
3
arrow_downward

Это должно быть довольно эффективным -

%// Case1 and case 2 matches
case1_matches = a1(1:end-1,:) == a1(2:end,:) & a1(1:end-1,:)==1
case2_matches = a1(:,1:end-1) == a1(:,2:end) & a1(:,1:end-1)==1

%// Get sum of those matches for the final output, equivalent to your pc
out = sum(case1_matches(:)) + sum(case2_matches(:))

Вы можете заменить sum(..(:)) на nnz(), но я сомневаюсь, что это будет лучше, чем sum с точки зрения производительности во время выполнения в соответствии с контрольные показатели sum по сравнению с nnz.

person Divakar    schedule 21.01.2015