diff --git a/modules/ml/src/em.cpp b/modules/ml/src/em.cpp index d5b2e29b81..72617f9e53 100644 --- a/modules/ml/src/em.cpp +++ b/modules/ml/src/em.cpp @@ -450,7 +450,9 @@ bool EM::doTrain(int startStep, OutputArray labels, OutputArray probs, OutputArr setIdentity(covs[clusterIndex], Scalar(covsEigenValues[clusterIndex].at(0))); } else if(covMatType == EM::COV_MAT_DIAGONAL) - covs[clusterIndex] = Mat::diag(covsEigenValues[clusterIndex].t()); + { + covs[clusterIndex] = Mat::diag(covsEigenValues[clusterIndex]); + } } if(labels.needed())