From 917561bb421bb7f4e790f49f43e75999bb9f0bf6 Mon Sep 17 00:00:00 2001 From: harakas Date: Sun, 28 Jan 2024 00:45:20 +0200 Subject: [PATCH] docker/rocm: update README.md --- docker/rocm/README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docker/rocm/README.md b/docker/rocm/README.md index f81bda733..fe4540b2c 100644 --- a/docker/rocm/README.md +++ b/docker/rocm/README.md @@ -57,6 +57,8 @@ The `*-frigate.txt` labels are grouped to a few relevant categories one would li I have also implemented a CPU detector based on the ONNX engine. So you can run the models without the GPU, just change the `detector.rocm.type` into `onnx`. +The models are added into the docker images automatically, they are taken from my [models](https://github.com/harakas/models/tree/main/ultralytics/yolov8.1) repository. + ## Performance On an AMD Ryzen 3 5400U with Radeon Graphics ([ASRock 4X4 BOX-5400U](https://www.asrockind.com/en-gb/4X4%20BOX-5400U)) I'm getting about 140fps detections with yolov8n and 65fps with yolov8s.