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

Add GradCAM for object detection #75

Merged
merged 39 commits into from
May 22, 2023
Merged

Add GradCAM for object detection #75

merged 39 commits into from
May 22, 2023

Commits on Mar 24, 2023

  1. Add example with GradCAM for YOLOv5 in object detection 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Mar 24, 2023
    Configuration menu
    Copy the full SHA
    bb9aace View commit details
    Browse the repository at this point in the history

Commits on Mar 27, 2023

  1. Remove constraints for min and max bbox size 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Mar 27, 2023
    Configuration menu
    Copy the full SHA
    95ebde4 View commit details
    Browse the repository at this point in the history

  2. Extract base ObjectDetector class 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Mar 27, 2023
    Configuration menu
    Copy the full SHA
    800d316 View commit details
    Browse the repository at this point in the history

  3. Remove previous example with object detection task 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Mar 27, 2023
    Configuration menu
    Copy the full SHA
    7bf25d7 View commit details
    Browse the repository at this point in the history

  4. Add original code source URL in docstring 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Mar 27, 2023
    Configuration menu
    Copy the full SHA
    0e06077 View commit details
    Browse the repository at this point in the history

  5. Move fetching layer from YOLO model function, add target_layer parame… 

    …ter to GradCAM

Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Mar 27, 2023
    Configuration menu
    Copy the full SHA
    3cb9096 View commit details
    Browse the repository at this point in the history

  6. Add dataclasses for complex object detection output data types 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Mar 27, 2023
    Configuration menu
    Copy the full SHA
    b0a2f72 View commit details
    Browse the repository at this point in the history

Commits on Mar 30, 2023

  1. Refactor code, add docstrings, fix names, add workaround with np.abs … 

    …for bbox coordinates

Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Mar 30, 2023
    Configuration menu
    Copy the full SHA
    a109c95 View commit details
    Browse the repository at this point in the history

Commits on Apr 3, 2023

  1. Add support for torchvision SSD model for object detection, refactoring 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 3, 2023
    Configuration menu
    Copy the full SHA
    729b160 View commit details
    Browse the repository at this point in the history

  2. Rename directories 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 3, 2023
    Configuration menu
    Copy the full SHA
    cba9096 View commit details
    Browse the repository at this point in the history

  3. Refactor, change BaseObjectDetector class definition, make SSD model … 

    …inherit from it

Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 3, 2023
    Configuration menu
    Copy the full SHA
    c668ea0 View commit details
    Browse the repository at this point in the history

Commits on Apr 4, 2023

  1. Refactor modules structure, add GradCAM to object detection module 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 4, 2023
    Configuration menu
    Copy the full SHA
    4c64201 View commit details
    Browse the repository at this point in the history

  2. Remove unused imports 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 4, 2023
    Configuration menu
    Copy the full SHA
    80675b6 View commit details
    Browse the repository at this point in the history

  3. Refactor, move OD models to separate module, fix YOLO prediction gene… 

    …ration

Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 4, 2023
    Configuration menu
    Copy the full SHA
    5202503 View commit details
    Browse the repository at this point in the history

  4. Apply pre-commit hooks to all files 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 4, 2023
    Configuration menu
    Copy the full SHA
    f07bddc View commit details
    Browse the repository at this point in the history

Commits on Apr 5, 2023

  1. Replace excessive unnecessary dependency with simple parsing 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 5, 2023
    Configuration menu
    Copy the full SHA
    6c4d599 View commit details
    Browse the repository at this point in the history

  2. Remove model warmup run 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 5, 2023
    Configuration menu
    Copy the full SHA
    1cf27ba View commit details
    Browse the repository at this point in the history

  3. Remove unused path in forward pass algorithm 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 5, 2023
    Configuration menu
    Copy the full SHA
    3d628be View commit details
    Browse the repository at this point in the history

Commits on Apr 6, 2023

  1. Replace custom implementations with torchvision imports 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 6, 2023
    Configuration menu
    Copy the full SHA
    9cc543e View commit details
    Browse the repository at this point in the history

Commits on Apr 7, 2023

  1. Add unit tests for object detection utils 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 7, 2023
    Configuration menu
    Copy the full SHA
    2a3654b View commit details
    Browse the repository at this point in the history

  2. Remove obsolete directory 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 7, 2023
    Configuration menu
    Copy the full SHA
    74f3af3 View commit details
    Browse the repository at this point in the history

Commits on Apr 11, 2023

  1. Fix YOLOv5 bbox conversion 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 11, 2023
    Configuration menu
    Copy the full SHA
    dba4220 View commit details
    Browse the repository at this point in the history

  2. Add unit test for object detection visualization utils, refactor 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 11, 2023
    Configuration menu
    Copy the full SHA
    e8ca464 View commit details
    Browse the repository at this point in the history

  3. Refactor GradCAM for OD 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 11, 2023
    Configuration menu
    Copy the full SHA
    615b29f View commit details
    Browse the repository at this point in the history

Commits on Apr 26, 2023

  1. Remove redundant device argument to YOLOv5ObjectDetector class initia… 

    …lizer

Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 26, 2023
    Configuration menu
    Copy the full SHA
    74bd96a View commit details
    Browse the repository at this point in the history

  2. Fix unit tests for image preprocessing - use rectangle instead of squ… 

    …are shapes

Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 26, 2023
    Configuration menu
    Copy the full SHA
    671a6a6 View commit details
    Browse the repository at this point in the history

  3. Simplify forward function in WrapperYOLOv5ObjectDetectionModule class 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed Apr 26, 2023
    Configuration menu
    Copy the full SHA
    e9caa21 View commit details
    Browse the repository at this point in the history

Commits on May 9, 2023

  1. Add custom GradCAM base algorithm implementation for classification a… 

    …nd object detection

Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 9, 2023
    Configuration menu
    Copy the full SHA
    2ef603e View commit details
    Browse the repository at this point in the history

Commits on May 16, 2023

  1. Move object detection model examples to examples directory 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 16, 2023
    Configuration menu
    Copy the full SHA
    c716d68 View commit details
    Browse the repository at this point in the history

Commits on May 17, 2023

  1. Restructure library directory structure for explainer algorithms, mov… 

    …e object detection example script to notebook

Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 17, 2023
    Configuration menu
    Copy the full SHA
    3de8d7c View commit details
    Browse the repository at this point in the history

  2. Add interpolation method parameter to resize_image function 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 17, 2023
    Configuration menu
    Copy the full SHA
    b6adfe5 View commit details
    Browse the repository at this point in the history

  3. Fix unit tests and imports 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 17, 2023
    Configuration menu
    Copy the full SHA
    77fa17a View commit details
    Browse the repository at this point in the history

  4. Remove adding epsilon in preprocess object detection image function 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 17, 2023
    Configuration menu
    Copy the full SHA
    bc0b278 View commit details
    Browse the repository at this point in the history

  5. Fix example notebook - add assertions of YOLOv5 image shape 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 17, 2023
    Configuration menu
    Copy the full SHA
    cc968da View commit details
    Browse the repository at this point in the history

  6. Fix basic usage notebook after refactoring class names and directory … 

    …structure

Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 17, 2023
    Configuration menu
    Copy the full SHA
    581c4f4 View commit details
    Browse the repository at this point in the history

Commits on May 18, 2023

  1. Enable changing image ratio to match YOLOv5 requirements of image shape 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 18, 2023
    Configuration menu
    Copy the full SHA
    734dd22 View commit details
    Browse the repository at this point in the history

Commits on May 19, 2023

  1. Update README 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 19, 2023
    Configuration menu
    Copy the full SHA
    18f8c40 View commit details
    Browse the repository at this point in the history

  2. Refactor object detection custom modules 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 19, 2023
    Configuration menu
    Copy the full SHA
    4d811a1 View commit details
    Browse the repository at this point in the history

  3. Refactor GradCAM for object detection 

    Signed-off-by: Adam Wawrzyński <adam.wawrzynski@reasonfieldlab.com>
    Adam Wawrzyński committed May 19, 2023
    Configuration menu
    Copy the full SHA
    b146038 View commit details
    Browse the repository at this point in the history