Add check and scale function (unfinished)

This commit is contained in:
kradchen
2023-05-31 13:55:39 +08:00
parent a82b89966a
commit 4270a11ea4
4 changed files with 31 additions and 4 deletions

View File

@@ -21,7 +21,7 @@ namespace Recon
bool nonNeg = false;
};
Aurora::Matrix solve( Aurora::Sparse& M, const Aurora::Matrix& b, const Aurora::Matrix& dims, int niter, TVAL3SolverOptions solverOptions){
Aurora::Matrix solve( Aurora::Sparse& M, Aurora::Matrix& b, const Aurora::Matrix& dims, int niter, TVAL3SolverOptions solverOptions){
if (Recon::transParams::name.empty()){
Recon::transParams::name = "TVAL3";
}
@@ -82,6 +82,7 @@ namespace Recon
options.TVAL3Beta0 = Recon::transParams::betaValues[i];
options.nonNeg = nonNeg;
solveResult[j] = solve(M, b, dims, transParams::maxIter, options);
solveResult[j].forceReshape(dims[0], dims[1], dims.getDataSize()<3?1:dims[2]);
}
result[i] = solveResult;
}