zx377262372 / dockerfile-yolov5-jetson

Dockerfile for yolov5 inference on NVIDIA Jetson

this repository

This repository builds docker image for object detection using Yolov5 on Nvidia Jetson platform. All operations below should be done on Jetson platform.

build

This operation build docker image named 'yolov5'.

./build.sh

If the build fails, use --no-cache option to clean the docker build cache.

./build.sh --no-cache

run

This operation detects objects with camera connected to /dev/video0.

./run.sh

run with your own weights

You can use your own weights(my-weights.pt), as follows:

mkdir -p /path/to/weights
cp my-weights.pt /path/to/weights
xhost +local:
docker run -it --rm \
           --runtime nvidia \
           --network host \
           --device /dev/video0:/dev/video0:mrw \
           -e DISPLAY=$DISPLAY \
           -e LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libgomp.so.1 \
           -v /tmp/.X11-unix/:/tmp/.X11-unix \
           -v /path/to/weights:/weights \
           yolov5 python3.8 detect.py --source 0 --weights /weights/my-weights.pt

CSI Camera support

CSI Camera is not supported for this docker environment.