Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

AMD GPU support with the rocm detector and YOLOv8 pretrained model download and support for edgetpu and openvino detectors #9760

Closed
wants to merge 46 commits into from
Closed

AMD GPU support with the rocm detector and YOLOv8 pretrained model download and support for edgetpu and openvino detectors #9760

wants to merge 46 commits into from

Commits on Jan 25, 2024

  1. ROCm AMD/GPU based build and detector, WIP

    @harakas
    harakas committed Jan 25, 2024
    Configuration menu
    Copy the full SHA
    42f1168 View commit details
    Browse the repository at this point in the history

Commits on Jan 26, 2024

  1. detectors/rocm: separate yolov8 postprocessing into own function; fix… 

    … box scaling; use cv2.dnn.blobForImage for preprocessing; assert on required model parameters
    @harakas
    harakas committed Jan 26, 2024
    Configuration menu
    Copy the full SHA
    d88accf View commit details
    Browse the repository at this point in the history

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

    @harakas
    harakas committed Jan 26, 2024
    Configuration menu
    Copy the full SHA
    35ae69a View commit details
    Browse the repository at this point in the history

  3. docker/rocm: make imported model files readable by all

    @harakas
    harakas committed Jan 26, 2024
    Configuration menu
    Copy the full SHA
    fb2425e View commit details
    Browse the repository at this point in the history

  4. docker/rocm: readme about running on AMD GPUs

    @harakas
    harakas committed Jan 26, 2024
    Configuration menu
    Copy the full SHA
    bb0836b View commit details
    Browse the repository at this point in the history

  5. docker/rocm: updated README

    @harakas
    harakas authored Jan 26, 2024
    Configuration menu
    Copy the full SHA
    60ea2a4 View commit details
    Browse the repository at this point in the history

  6. docker/rocm: updated README

    @harakas
    harakas authored Jan 26, 2024
    Configuration menu
    Copy the full SHA
    6de14a4 View commit details
    Browse the repository at this point in the history

  7. docker/rocm: updated README

    @harakas
    harakas authored Jan 26, 2024
    Configuration menu
    Copy the full SHA
    c93a508 View commit details
    Browse the repository at this point in the history

Commits on Jan 27, 2024

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

    @harakas
    harakas committed Jan 27, 2024
    Configuration menu
    Copy the full SHA
    622a45e View commit details
    Browse the repository at this point in the history

  2. detector/plugins: added onnx cpu plugin

    @harakas
    harakas committed Jan 27, 2024
    Configuration menu
    Copy the full SHA
    af3e40d View commit details
    Browse the repository at this point in the history

  3. docker/rocm: updated container with limite label sets

    @harakas
    harakas committed Jan 27, 2024
    Configuration menu
    Copy the full SHA
    252a5dd View commit details
    Browse the repository at this point in the history

  4. example detectors view

    @harakas
    harakas committed Jan 27, 2024
    Configuration menu
    Copy the full SHA
    4481d08 View commit details
    Browse the repository at this point in the history

  5. docker/rocm: updated README.md

    @harakas
    harakas authored Jan 27, 2024
    Configuration menu
    Copy the full SHA
    3747b8e View commit details
    Browse the repository at this point in the history

  6. docker/rocm: update README.md

    @harakas
    harakas authored Jan 27, 2024
    Configuration menu
    Copy the full SHA
    917561b View commit details
    Browse the repository at this point in the history

Commits on Jan 28, 2024

  1. docker/rocm: do not set HSA_OVERRIDE_GFX_VERSION at all for the gener… 

    …al version as the empty value broke rocm
    @harakas
    harakas committed Jan 28, 2024
    Configuration menu
    Copy the full SHA
    2634348 View commit details
    Browse the repository at this point in the history

  2. detectors: simplified/optimized yolov8_postprocess

    @harakas
    harakas committed Jan 28, 2024
    Configuration menu
    Copy the full SHA
    de02ba7 View commit details
    Browse the repository at this point in the history

  3. detector/yolo_utils: indentation, remove unused variable

    @harakas
    harakas committed Jan 28, 2024
    Configuration menu
    Copy the full SHA
    da31ee1 View commit details
    Browse the repository at this point in the history

Commits on Jan 31, 2024

  1. detectors/rocm: default option to conserve cpu usage at the expense o… 

    …f latency
    @harakas
    harakas committed Jan 31, 2024
    Configuration menu
    Copy the full SHA
    6171311 View commit details
    Browse the repository at this point in the history

Commits on Feb 2, 2024

  1. detectors/yolo_utils: use nms to prefilter overlapping boxes if too m… 

    …any detected
    @harakas
    harakas committed Feb 2, 2024
    Configuration menu
    Copy the full SHA
    cd50898 View commit details
    Browse the repository at this point in the history

  2. detectors/edgetpu_tfl: add support for yolov8

    @harakas
    harakas committed Feb 2, 2024
    Configuration menu
    Copy the full SHA
    80ae30e View commit details
    Browse the repository at this point in the history

Commits on Feb 4, 2024

  1. util/download_models: script to download yolov8 model files

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    d6ff220 View commit details
    Browse the repository at this point in the history

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

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    7c58c3e View commit details
    Browse the repository at this point in the history

  3. detectors/rocm: assume models are in /config/model_cache/yolov8/

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    b151bcc View commit details
    Browse the repository at this point in the history

  4. docker/rocm: compile onnx files into mxr files at startup

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    8a14703 View commit details
    Browse the repository at this point in the history

  5. switch model download into bash script

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    77fc38c View commit details
    Browse the repository at this point in the history

  6. detectors/rocm: automatically override HSA_OVERRIDE_GFX_VERSION for c… 

    …ouple of known chipsets
    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    24cae6d View commit details
    Browse the repository at this point in the history

  7. docs: rocm detector first notes

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    1942783 View commit details
    Browse the repository at this point in the history

  8. typos

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    c037c18 View commit details
    Browse the repository at this point in the history

  9. describe builds (harakas temporary)

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    80bb349 View commit details
    Browse the repository at this point in the history

  10. docker/rocm: also build a version for gfx1100

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    d35a2f1 View commit details
    Browse the repository at this point in the history

  11. docker/rocm: use cp instead of tar

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    b2d26c9 View commit details
    Browse the repository at this point in the history

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

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    4bcb071 View commit details
    Browse the repository at this point in the history

  13. frigate/detectors: renamed yolov8_preprocess->preprocess, pass input … 

    …tensor element type
    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    a982698 View commit details
    Browse the repository at this point in the history

  14. docker/main: use newer openvino (2023.3.0)

    @harakas
    harakas committed Feb 4, 2024
    Configuration menu
    Copy the full SHA
    dee95de View commit details
    Browse the repository at this point in the history

Commits on Feb 5, 2024

  1. detectors: implement class aggregation

    @harakas
    harakas committed Feb 5, 2024
    Configuration menu
    Copy the full SHA
    dcfe6bb View commit details
    Browse the repository at this point in the history

Commits on Feb 6, 2024

  1. update yolov8 model

    @harakas
    harakas committed Feb 6, 2024
    Configuration menu
    Copy the full SHA
    c2baac6 View commit details
    Browse the repository at this point in the history

  2. add openvino/yolov8 support for label aggregation

    @harakas
    harakas committed Feb 6, 2024
    Configuration menu
    Copy the full SHA
    35cc6a1 View commit details
    Browse the repository at this point in the history

Commits on Feb 7, 2024

  1. docker: remove pointless s6/timeout-up files

    @harakas
    harakas committed Feb 7, 2024
    Configuration menu
    Copy the full SHA
    d117da4 View commit details
    Browse the repository at this point in the history

Commits on Feb 9, 2024

  1. Revert "detectors: implement class aggregation" 

    This reverts commit dcfe6bb.
    @harakas
    harakas committed Feb 9, 2024
    Configuration menu
    Copy the full SHA
    4dc7502 View commit details
    Browse the repository at this point in the history

  2. detectors/openvino: remove class aggregation

    @harakas
    harakas committed Feb 9, 2024
    Configuration menu
    Copy the full SHA
    79b88c9 View commit details
    Browse the repository at this point in the history

  3. detectors: increase yolov8 postprocessing score trershold to 0.5

    @harakas
    harakas committed Feb 9, 2024
    Configuration menu
    Copy the full SHA
    4318b5a View commit details
    Browse the repository at this point in the history

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

    @harakas
    harakas committed Feb 9, 2024
    Configuration menu
    Copy the full SHA
    a00fe53 View commit details
    Browse the repository at this point in the history

  5. Update object_detectors.md

    @harakas
    harakas authored Feb 9, 2024
    Configuration menu
    Copy the full SHA
    7b2331b View commit details
    Browse the repository at this point in the history

  6. Merge branch 'blakeblackshear:dev' into dev

    @harakas
    harakas authored Feb 9, 2024
    Configuration menu
    Copy the full SHA
    d09102a View commit details
    Browse the repository at this point in the history

  7. updated CODEOWNERS file for rocm

    @harakas
    harakas committed Feb 9, 2024
    Configuration menu
    Copy the full SHA
    86cf8fa View commit details
    Browse the repository at this point in the history

  8. updated build names for documentation

    @harakas
    harakas committed Feb 9, 2024
    Configuration menu
    Copy the full SHA
    7237a79 View commit details
    Browse the repository at this point in the history