search

burningion / bicyclist-defense-jetson

Defending Bicyclists from Erratic Drivers with Computer Vision and mmWave Radar
https://www.makeartwithpython.com/blog/building-a-data-pipeline-for-robotics/
7 stars 0 forks source link

TODO: Distill models using recorded training data #1

Open burningion opened 5 months ago

burningion commented 5 months ago

Use captured bicycle video recordings to distill our ViT and SAM models used for inference.

burningion commented 5 months ago

Looking at the gst-nvtracker page, I found a link to TrafficCamNet, which should give us decent results right out of the box.

Going to try downloading the model and creating a TensorRT to see what kind of performance I can get out of the Jetson Orin Nano.

burningion commented 5 months ago

Added the TensorRT model to models/, created TensorRT model by running the repo container, then running:

$ /usr/src/tensorrt/bin/trtexec --onnx=resnet18_trafficcamnet_pruned.onnx --saveEngine=resnet18_trafficcamera_pruned.engine

I did this in the directory I copied the source model to.