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.