71 lines
3.1 KiB
PowerShell
71 lines
3.1 KiB
PowerShell
param(
|
|
[Parameter(Mandatory = $true)][string]$Prepared,
|
|
[Parameter(Mandatory = $true)][string]$OutputRoot,
|
|
[string]$ReferencePoseFile = "reference_poses_rtk_gga_raw_heading.csv",
|
|
[double]$ReferenceHeight = 0.8535,
|
|
[int]$MinPairs = 20,
|
|
[int]$Bootstrap = 100
|
|
)
|
|
|
|
$ErrorActionPreference = "Stop"
|
|
$Repo = Split-Path -Parent $PSScriptRoot
|
|
$Code = Join-Path $Repo "code\rigorous_calibration.py"
|
|
$Refine = Join-Path $Repo "code\refine_pairs.py"
|
|
$Consensus = Join-Path $Repo "code\cross_backend_filter.py"
|
|
$Frames = Join-Path $Prepared "frames_all"
|
|
$ReferencePoses = Join-Path $Prepared $ReferencePoseFile
|
|
$Common = Join-Path $OutputRoot "common"
|
|
$Open = Join-Path $OutputRoot "open3d_gicp"
|
|
$Small = Join-Path $OutputRoot "small_gicp"
|
|
$ConsensusOut = Join-Path $OutputRoot "consensus"
|
|
|
|
function Run-Python {
|
|
param([string]$Stage, [string[]]$Arguments)
|
|
Write-Host "[$Stage]"
|
|
& python @Arguments
|
|
if ($LASTEXITCODE -ne 0) { throw "$Stage failed with Python exit code $LASTEXITCODE" }
|
|
}
|
|
|
|
foreach ($Path in @($Frames, $ReferencePoses)) {
|
|
if (-not (Test-Path -LiteralPath $Path)) { throw "Input does not exist: $Path" }
|
|
}
|
|
New-Item -ItemType Directory -Force -Path $Common,$Open,$Small,$ConsensusOut | Out-Null
|
|
|
|
$Ground = Join-Path $Common "ground_planes.csv"
|
|
Run-Python "ground planes" @($Code, "ground", "--frames", $Frames, "--output", $Ground)
|
|
|
|
foreach ($Backend in @("small_gicp", "open3d")) {
|
|
$Directory = if ($Backend -eq "small_gicp") { $Small } else { $Open }
|
|
$Raw = Join-Path $Directory "B_estimation.npz"
|
|
$QualityJson = Join-Path $Directory "B_quality.json"
|
|
$QualityCsv = Join-Path $Directory "B_quality.csv"
|
|
$PairArgs = @($Code, "pairs", "--backend", $Backend, "--frames", $Frames, "--reference-poses", $ReferencePoses,
|
|
"--output", $Raw, "--quality-json", $QualityJson, "--quality-csv", $QualityCsv,
|
|
"--min-pairs", "$MinPairs")
|
|
if ($Backend -eq "open3d") { $PairArgs += @("--max-gap", "3", "--multistart", "1", "--iterations", "40") }
|
|
Run-Python "$Backend pairs" $PairArgs
|
|
Run-Python "$Backend X-independent refinement" @(
|
|
$Refine, "--pairs", $Raw, "--quality-json", $QualityJson,
|
|
"--output", (Join-Path $Directory "B_refined.npz"), "--min-pairs", "$MinPairs"
|
|
)
|
|
Run-Python "$Backend calibration" @(
|
|
$Code, "calibrate", "--pairs", (Join-Path $Directory "B_refined.npz"),
|
|
"--ground-planes", $Ground, "--reference-height", "$ReferenceHeight",
|
|
"--bootstrap", "$Bootstrap", "--output", (Join-Path $Directory "extrinsic.json")
|
|
)
|
|
}
|
|
|
|
$ConsensusPairs = Join-Path $ConsensusOut "B_consensus.npz"
|
|
Run-Python "cross-backend consensus" @(
|
|
$Consensus, "--open3d-pairs", (Join-Path $Open "B_refined.npz"),
|
|
"--small-pairs", (Join-Path $Small "B_refined.npz"),
|
|
"--output", $ConsensusPairs, "--min-pairs", "$MinPairs"
|
|
)
|
|
Run-Python "consensus calibration" @(
|
|
$Code, "calibrate", "--pairs", $ConsensusPairs, "--ground-planes", $Ground,
|
|
"--reference-height", "$ReferenceHeight", "--bootstrap", "$Bootstrap",
|
|
"--output", (Join-Path $ConsensusOut "extrinsic.json")
|
|
)
|
|
|
|
Write-Host "Calibration results: $OutputRoot"
|