Convert nano model into TensorRT for Jetson Nano

Salmon-Computer-Vision / salmon-computer-vision

MIT License

8 stars 1 forks source link

Convert nano model into TensorRT for Jetson Nano #76

Open KamiCreed opened 7 months ago

KamiCreed commented 7 months ago

As we are using YOLOv8, there are likely some documentation on how to deploy a nano model to something like a Jetson Nano. Probably conversion to TensorRT.

Check the note in utils/jetson/README.md on the Jetson Nano setup.

YOLOv8n model is in this full dataset Dropbox link.

Salmon counting code is in training/pysalmcount/pysalmcount/salmon_counter.py

KamiCreed commented 3 months ago

YOLOv8 aka Ultralytics finally have a docker image for Jetpack 4 aka the old Jetson Nanos through ultralytics/ultralytics:latest-jetson-jetpack4 on the docker hub. Can be used to compile TensorRT and run the inferencing on a 4GB Jetson nano.