dustynv/nanosam

nanosam Docker镜像文档

镜像概述

nanosam Docker镜像基于NVIDIA-AI-IOT的https://github.com/NVIDIA-AI-IOT/nanosam/%E6%9E%84%E5%BB%BA%EF%BC%8C%E4%B8%93%E4%B8%BAJetson%E8%AE%BE%E5%A4%87%E4%BC%98%E5%8C%96%EF%BC%8C%E6%8F%90%E4%BE%9B%E9%AB%98%E6%95%88%E7%9A%84%E5%9B%BE%E5%83%8F%E5%88%86%E5%89%B2%E8%83%BD%E5%8A%9B%E3%80%82%E8%AF%A5%E9%95%9C%E5%83%8F%E9%9B%86%E6%88%90%E4%BA%86%E5%BF%85%E8%A6%81%E7%9A%84%E4%BE%9D%E8%B5%96%E7%BB%84%E4%BB%B6%EF%BC%8C%E5%8F%AF%E7%9B%B4%E6%8E%A5%E7%94%A8%E4%BA%8E%E8%BF%90%E8%A1%8C%E5%9B%BE%E5%83%8F%E5%88%86%E5%89%B2%E4%BB%BB%E5%8A%A1%E3%80%81%E6%89%A7%E8%A1%8C%E5%9F%BA%E5%87%86%E6%B5%8B%E8%AF%95%E5%8F%8A%E7%9B%B8%E5%85%B3%E5%BC%80%E5%8F%91%E5%B7%A5%E4%BD%9C%E3%80%82

核心功能与特性

• 图像分割：支持基于预训练模型的图像分割任务
• 基准测试：提供性能评估工具，可生成测试报告和结果图像
• Jetson优化：针对Jetson平台（L4T系统）进行优化，支持GPU加速
• 完整依赖：集成CUDA、PyTorch、TensorRT等深度学习框架及工具链

依赖项

• 系统要求：L4T ['>=34.1.0']（JetPack 5.1+）
• 核心依赖组件：
  • build-essential、cmake、rust等构建工具
  • cuda、cudnn、tensorrt等GPU加速库
  • python、numpy等基础运行环境
  • pytorch:2.2、torchvision、torch2trt等深度学习框架
  • onnx、huggingface_hub、transformers等模型相关工具

容器镜像信息

兼容性说明：

• L4T R32.7容器可在其他L4T R32.7版本（JetPack 4.6+）上运行
• L4T R35.x容器可在其他L4T R35.x版本（JetPack 5.1+）上运行

使用方法

基本使用示例

运行基本图像分割示例并将结果复制到主机：

bash
./run.sh $(./autotag nanosam) /bin/bash -c " \
  cd /opt/nanosam && \
  python3 examples/basic_usage.py \  
    --image_encoder=data/resnet18_image_encoder.engine \
    --mask_decoder=data/mobile_sam_mask_decoder.engine && \
  mv data/basic_usage_out.jpg /data/ \
  "

基准测试

执行性能基准测试并生成报告：

bash
 ./run.sh $(./autotag nanosam) /bin/bash -c " \
   cd /opt/nanosam && \
   python3 benchmark.py --run 3 -s /data/nanosam.csv && \
   mv data/benchmark_last_image.jpg /data/ \
   "

运行容器

使用jetson-containers工具（推荐）

bash
# 自动拉取或构建兼容的容器镜像
jetson-containers run $(autotag nanosam)

# 显式指定镜像版本
jetson-containers run dustynv/nanosam:r35.2.1

使用docker run命令

bash
sudo docker run --runtime nvidia -it --rm --network=host dustynv/nanosam:r35.2.1

挂载主机目录

bash
jetson-containers run -v /host/path:/container/path $(autotag nanosam)

运行指定命令

bash
jetson-containers run $(autotag nanosam) python3 examples/basic_usage.py --help

构建容器

如需手动构建镜像，先完成https://github.com/dusty-nv/jetson-containers/tree/master/docs/setup.md%EF%BC%8C%E7%84%B6%E5%90%8E%E6%89%A7%E8%A1%8C%EF%BC%9A

bash
jetson-containers build nanosam

可通过--help参数查看构建选项：

bash
jetson-containers build nanosam --help

Dockerfile

镜像构建文件：https://github.com/dusty-nv/jetson-containers/tree/master/packages/vit/nanosam/Dockerfile