frigate/converters/yolo4/build.sh

18 lines
516 B
Bash
Executable File

#!/bin/bash
mkdir -p $(pwd)/model
mkdir -p $(pwd)/plugin
docker build --tag models.yolo4 --file ./Dockerfile.l4t.tf15 ./assets/
sudo docker run --rm -it --name models.yolo4 \
--mount type=tmpfs,target=/tmp/cache,tmpfs-size=1000000000 \
-v $(pwd)/model:/model:rw \
-v $(pwd)/plugin:/plugin:rw \
-v /tmp/argus_socket:/tmp/argus_socket \
-e NVIDIA_VISIBLE_DEVICES=all \
-e NVIDIA_DRIVER_CAPABILITIES=compute,utility,video \
--runtime=nvidia \
--privileged \
models.yolo4 /run.sh