在 Jetson Nano 上运行自定义训练的 yolov8 模型

Question

我自定义训练了一个 yolov8n 模型，我想在 Jetson Nano 上运行它。有人设法做到这一点吗？如果是的话，你能帮我一下吗？

我有一个自定义训练模型的 .pt。 NVidia 在其网站上提供的操作系统映像是 Ubuntu 18.04，我遇到了许多兼容性问题。我很想知道是否有人设法让 yolo 在 Jetson 上运行，特别是 ultralytics 的 yolov8n 模型。

提前致谢！

Answer 1

我成功地使用 Qengineering 的非官方 Ubuntu 20.04 映像在 Jetson Nano 中运行 YOLOv8n 检测模型。

之后，您只需要安装Ultralytics并照常运行YOLO模型即可。

但是，根据我的经验，Jetson Nano 中的模型存在问题。即使使用 CLI (

yolo predict ...

)，运行模型的程序也会使用高内存 (2GB+)。所以，除非在 Jetson Nano 中运行检测程序，否则我什么也做不了。

总之，它有效，但又带来了另一个问题。让我们看看您或其他人是否有同样的问题。