NVIDIA-AI-IOT / redtail

Perception and AI components for autonomous mobile robotics.
BSD 3-Clause "New" or "Revised" License
1.01k stars 346 forks source link

Regarding tensorRT of TrailNet_SResNet-18 #152

Open MSP93 opened 4 years ago

MSP93 commented 4 years ago

Is the TrailNet_SResNet-18 Pretrained model an optimized ones that built using TensorRT? If so can you please share pretrained model without optimized ones.

Alexey-Kamenev commented 4 years ago

TrailNet Caffe model was originally trained in Caffe using FP32 precision. The model is used by our ROS node which uses TensorRT to load and optimize it for fast inference. The target precision FP16/FP32 is selected based on user-provided parameters, see launch files for more details