azaion/detections
1
0
Fork 0
mirror of https://github.com/azaion/detections.git synced 2026-06-21 10:11:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Changed to use prepare for TensorRT
ci/woodpecker/push/02-build-push Pipeline was successful
Details

Browse Source
This commit is contained in:
Roman Meshko
2026-05-23 22:07:01 +03:00
parent a85c238c92
commit bd9bc2e05d
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/jetson/generate_int8_cache.py
+7
View File
@@ -98,6 +98,13 @@ def main():
print(f"Cache written → {args.output}") print(f"Cache written → {args.output}")
onnx_data = Path(args.onnx).read_bytes() onnx_data = Path(args.onnx).read_bytes()
try:
from engines.onnx_tensorrt_compat import prepare_for_tensorrt
onnx_data = prepare_for_tensorrt(onnx_data)
print("Prepared ONNX model for TensorRT static Jetson build")
except Exception as e:
print(f"WARNING: ONNX TensorRT compatibility preparation failed: {e}", file=sys.stderr)
logger = trt.Logger(trt.Logger.INFO) logger = trt.Logger(trt.Logger.INFO)
explicit_batch = 1 << int(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH) explicit_batch = 1 << int(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH)