nnicolosi/Raptor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add ablation study
Validate Operations / validate-operations (push) Has been cancelled
Details

Browse Source 
normalize names and artifact paths
This commit is contained in:
NiccoloN
2026-08-20 17:58:02 +02:00
parent add20e56eb
commit b009e1ff08
67 changed files with 1573 additions and 993 deletions
Download Patch File Download Diff File Expand all files Collapse all files
validation/tools/yolo_local_image_validation.py
+19 -15
View File
@@ -22,6 +22,7 @@ if sys.version_info < (3, 10):
)
from raptor_validation.onnx_utils import _ONNX_TO_NP, onnx_io, write_inputs_binary, write_inputs_to_memory_bin
from raptor_validation.artifacts import artifacts_dir
from raptor_validation.validate_one import (
MODE_COMPILE_ONLY,
build_dump_ranges,
@@ -64,16 +65,17 @@ def find_network_onnx(network_dir: Path) -> Path:
def local_case_paths(network_dir: Path, case_name: str):
artifact_root = artifacts_dir(network_dir)
return {
"root": network_dir,
"runner": network_dir / "runner" / "build" / "runner",
"runner_build": network_dir / "runner" / "build",
"raptor_pim": network_dir / "raptor" / "pim",
"real_root": network_dir / "real_image_validation",
"input_csv": network_dir / "real_image_validation" / "inputs" / f"{case_name}.csv",
"ref_dir": network_dir / "real_image_validation" / "reference" / case_name,
"sim_dir": network_dir / "real_image_validation" / "simulation" / case_name,
"sim_bin": network_dir / "real_image_validation" / "simulation" / case_name / "out.bin",
"root": artifact_root,
"runner": artifact_root / "runner" / "build" / "runner",
"runner_build": artifact_root / "runner" / "build",
"raptor_pim": artifact_root / "raptor" / "pim",
"real_root": artifact_root / "real_image_validation",
"input_csv": artifact_root / "real_image_validation" / "inputs" / f"{case_name}.csv",
"ref_dir": artifact_root / "real_image_validation" / "reference" / case_name,
"sim_dir": artifact_root / "real_image_validation" / "simulation" / case_name,
"sim_bin": artifact_root / "real_image_validation" / "simulation" / case_name / "out.bin",
}
@@ -102,10 +104,11 @@ def ensure_local_artifacts(args, network_onnx_path: Path):
def ensure_existing_artifacts(network_dir: Path):
artifact_root = artifacts_dir(network_dir)
required_paths = [
network_dir / "runner" / "build" / "runner",
network_dir / "raptor" / "pim" / "config.json",
network_dir / "raptor" / "pim" / "memory.bin",
artifact_root / "runner" / "build" / "runner",
artifact_root / "raptor" / "pim" / "config.json",
artifact_root / "raptor" / "pim" / "memory.bin",
]
missing = [str(path) for path in required_paths if not path.exists()]
if missing:
@@ -137,7 +140,7 @@ def run_local_reference_and_simulator(args, network_dir: Path, network_onnx_path
tensor = np.loadtxt(paths["input_csv"], delimiter=",", dtype=np.float32).reshape(1, 3, 640, 640)
write_inputs_to_memory_bin(paths["raptor_pim"] / "memory.bin", paths["raptor_pim"] / "config.json", [tensor])
input_bin = paths["sim_dir"] / "input.bin"
input_bin = paths["sim_dir"] / "input_0.bin"
write_inputs_binary(input_bin, [tensor])
dump_ranges = build_dump_ranges(paths["raptor_pim"] / "config.json", output_descriptors)
@@ -147,7 +150,8 @@ def run_local_reference_and_simulator(args, network_dir: Path, network_onnx_path
paths["sim_bin"],
dump_ranges,
timeout_sec=args.command_timeout_seconds,
input_paths=[input_bin],
input_dir=input_bin.parent,
batch_size=1,
)
return paths, output_descriptors[0]
@@ -225,7 +229,7 @@ def main():
parser.add_argument(
"--annotated-dir",
type=Path,
default=defaults["network_dir"] / "real_image_validation" / "annotated",
default=defaults["network_dir"] / "artifacts" / "real_image_validation" / "annotated",
)
args = parser.parse_args()