blakeblackshear.frigate/frigate
harakas 44d8cdbba1
AMD GPU support with the rocm detector and YOLOv8 pretrained model download (#9762)
* ROCm AMD/GPU based build and detector, WIP

* detectors/rocm: separate yolov8 postprocessing into own function; fix box scaling; use cv2.dnn.blobForImage for preprocessing; assert on required model parameters

* AMD/ROCm: add couple of more ultralytics models; comments

* docker/rocm: make imported model files readable by all

* docker/rocm: readme about running on AMD GPUs

* docker/rocm: updated README

* docker/rocm: updated README

* docker/rocm: updated README

* detectors/rocm: separated preprocessing functions into yolo_utils.py

* detector/plugins: added onnx cpu plugin

* docker/rocm: updated container with limite label sets

* example detectors view

* docker/rocm: updated README.md

* docker/rocm: update README.md

* docker/rocm: do not set HSA_OVERRIDE_GFX_VERSION at all for the general version as the empty value broke rocm

* detectors: simplified/optimized yolov8_postprocess

* detector/yolo_utils: indentation, remove unused variable

* detectors/rocm: default option to conserve cpu usage at the expense of latency

* detectors/yolo_utils: use nms to prefilter overlapping boxes if too many detected

* detectors/edgetpu_tfl: add support for yolov8

* util/download_models: script to download yolov8 model files

* docker/main: add download-models overlay into s6 startup

* detectors/rocm: assume models are in /config/model_cache/yolov8/

* docker/rocm: compile onnx files into mxr files at startup

* switch model download into bash script

* detectors/rocm: automatically override HSA_OVERRIDE_GFX_VERSION for couple of known chipsets

* docs: rocm detector first notes

* typos

* describe builds (harakas temporary)

* docker/rocm: also build a version for gfx1100

* docker/rocm: use cp instead of tar

* docker.rocm: remove README as it is now in detector config

* frigate/detectors: renamed yolov8_preprocess->preprocess, pass input tensor element type

* docker/main: use newer openvino (2023.3.0)

* detectors: implement class aggregation

* update yolov8 model

* add openvino/yolov8 support for label aggregation

* docker: remove pointless s6/timeout-up files

* Revert "detectors: implement class aggregation"

This reverts commit dcfe6bbf6f.

* detectors/openvino: remove class aggregation

* detectors: increase yolov8 postprocessing score trershold to 0.5

* docker/rocm: separate rocm distributed files into its own build stage

* Update object_detectors.md

* updated CODEOWNERS file for rocm

* updated build names for documentation

* Revert "docker/main: use newer openvino (2023.3.0)"

This reverts commit dee95de908.

* reverrted openvino detector

* reverted edgetpu detector

* scratched rocm docs from any mention of edgetpu or openvino

* Update docs/docs/configuration/object_detectors.md

Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com>

* renamed frigate.detectors.yolo_utils.py -> frigate.detectors.util.py

* clarified rocm example performance

* Improved wording and clarified text

* Mentioned rocm detector for AMD GPUs

* applied ruff formating

* applied ruff suggested fixes

* docker/rocm: fix missing argument resulting in larger docker image sizes

* docs/configuration/object_detectors: fix links to yolov8 release files

---------

Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com>
2024-02-10 06:41:46 -06:00
..
comms Write a low resolution low fps stream from decoded frames (#8673) 2024-01-31 12:56:11 +00:00
detectors AMD GPU support with the rocm detector and YOLOv8 pretrained model download (#9762) 2024-02-10 06:41:46 -06:00
events Add support for filtering history page and add support for creating timeline entries for audio / custom events (#9034) 2024-01-31 12:56:11 +00:00
images Replace green screen with error message and force camera_fps to 0 (#4544) 2022-11-28 21:47:20 -06:00
motion Adjust motion calibration to be more dynamic (#8250) 2023-10-20 19:22:38 -04:00
output Fix birdseye memory leak (#8992) 2024-01-31 12:56:11 +00:00
ptz Small autotracking changes (#9571) 2024-02-02 06:23:14 -06:00
record Write a low resolution low fps stream from decoded frames (#8673) 2024-01-31 12:56:11 +00:00
test Remove rtmp (#8941) 2024-01-31 12:56:11 +00:00
track Adjust kalman filter used for norfair tracker (#9538) 2024-02-02 06:22:56 -06:00
util Improve graph using pandas (#9234) 2024-01-31 12:56:11 +00:00
__init__.py app container and config schema 2021-01-26 21:40:33 -06:00
__main__.py Add isort and ruff linter (#6575) 2023-05-29 05:31:17 -05:00
app.py Streamline live view (#9772) 2024-02-10 05:30:53 -07:00
config.py Web & ffmpeg bug fixes (#9525) 2024-02-01 06:44:10 -06:00
const.py Automatically detect hwaccel args (#9142) 2024-01-31 12:56:11 +00:00
ffmpeg_presets.py Automatically detect hwaccel args (#9142) 2024-01-31 12:56:11 +00:00
http.py Improve graph using pandas (#9234) 2024-01-31 12:56:11 +00:00
log.py Remove faster_fifo (#7181) 2023-07-16 07:42:56 -05:00
models.py Write a low resolution low fps stream from decoded frames (#8673) 2024-01-31 12:56:11 +00:00
mypy.ini update deps (#6093) 2023-04-15 14:17:59 -05:00
object_detection.py Remove faster_fifo (#7181) 2023-07-16 07:42:56 -05:00
object_processing.py Autotracking: better handle objects that touch frame edges (#9062) 2024-01-31 12:56:11 +00:00
plus.py add support for docker secrets (#8409) 2023-11-02 05:35:30 -05:00
stats.py Fix bug on bad storage stats read (#8275) 2023-10-22 13:35:19 -05:00
storage.py Ensure storage cleanup doesn't fail (#8531) 2023-11-07 23:21:43 +00:00
timeline.py Improve graph using pandas (#9234) 2024-01-31 12:56:11 +00:00
types.py Add MQTT topic for active autotracking (#8419) 2023-11-01 23:20:26 +00:00
video.py Write recording segments to cache with timezone info (#8468) 2023-11-05 14:30:29 -06:00
watchdog.py Cleanup and organize utils (#7033) 2023-07-06 09:28:50 -05:00