export/tests

July 29, 2026 · View on GitHub

test_export_wrappers.py: PyTorch-only shape tests for export/wrappers.py (runs in default CI with torch).
test_faster_rcnn_export_parity.py: Pre-NMS export vs full faster_rcnn_inference; optional Hub pretrained checkpoint smoke.
test_oriented_rcnn_export_parity.py: Pre-NMS export vs OrientedRCNN eval; optional Hub pretrained smoke; optional ONNX/ORT.
test_keras_detect_bundle.py: Keras save/reload + relocatable model.onnx + class_id_to_name key coercion (requires TensorFlow).
test_ort_runtime.py: ORT device/provider helpers; TF GPU hide when CUDA; optional real-tile CUDA smoke (skipped without CUDA EP / ONNX artifact).
test_postprocess.py: Score prefilter vs post-NMS filter parity.
test_export_onnx_optional.py: ONNX export + checker / ORT (skipped unless onnx and, for the ORT test, onnxruntime are installed).
test_export_cli.py: odet export-* help/import smoke + export_tf orchestration (mocked; no Makefile).

Run from repo root:

pytest export/tests/

CI runs pytest tests/ export/tests/. Wrapper/parity/CLI tests run without TF; Keras bundle tests skip if TensorFlow is missing.