workspace/lane_build.sh
#! /usr/bin/bash

TRTEXEC=/home/jarvis/lean/TensorRT-8.6.1.6/bin/trtexec

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/jarvis/lean/TensorRT-8.6.1.6/lib

# ${TRTEXEC} \
#   --onnx=laneatt.sim.onnx \
#   --minShapes=images:1x3x360x640 \
#   --optShapes=images:1x3x360x640 \
#   --maxShapes=images:8x3x360x640 \
#   --memPoolSize=workspace:2048 \
#   --saveEngine=laneatt.sim.FP16.trtmodel \
#   --fp16 \
#   > laneatt.log 2>&1

# ${TRTEXEC} \
#   --onnx=clrnet.sim.onnx \
#   --minShapes=images:1x3x320x800 \
#   --optShapes=images:1x3x320x800 \
#   --maxShapes=images:8x3x320x800 \
#   --memPoolSize=workspace:2048 \
#   --saveEngine=clrnet.sim.FP16.trtmodel \
#   --fp16 \
#   > clrnet.log 2>&1

# ${TRTEXEC} \
#   --onnx=clrnet.static.sim.onnx \
#   --memPoolSize=workspace:2048 \
#   --saveEngine=clrnet.static.sim.FP16.trtmodel \
#   --fp16 \
#   > clrnet.static.log 2>&1

${TRTEXEC} \
  --onnx=clrernet.sim.onnx \
  --minShapes=images:1x3x320x800 \
  --optShapes=images:1x3x320x800 \
  --maxShapes=images:8x3x320x800 \
  --memPoolSize=workspace:2048 \
  --saveEngine=clrernet.sim.FP16.trtmodel \
  --fp16 \
  > clrernet.log 2>&1

# ${TRTEXEC} \
#   --onnx=clrernet.static.sim.onnx \
#   --memPoolSize=workspace:2048 \
#   --saveEngine=clrernet.static.sim.FP16.trtmodel \
#   --fp16 \
#   > clrernet.static.log 2>&1