TRT a custom frozen-graph

SteveMacenski / jetson_nano_detection_and_tracking

Jetson Nano ML install scripts, automated optimization of robotics detection models, and filter-based tracking of detections

GNU Lesser General Public License v2.1

226 stars 66 forks source link

TRT a custom frozen-graph #6

Closed jonimoi closed 5 years ago

jonimoi commented 5 years ago

How would you TRT a custom frozen graph that has been retrained on a ssd network?

SteveMacenski commented 5 years ago

its a pretty easy oneliner

trt_graph = trt.create_inference_graph(
    input_graph_def=frozen_graph,
    outputs=output_names,
    max_batch_size=1,
    max_workspace_size_bytes=1 << 25,
    precision_mode='FP16',
    minimum_segment_size=50
)