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

Bugfix of model.export() to work correct with bs>1 #1551

Merged
merged 18 commits into from
Oct 23, 2023
Merged

Bugfix of model.export() to work correct with bs>1 #1551

merged 18 commits into from
Oct 23, 2023

Commits on Oct 18, 2023

  1. Bugfixes: 

     1) Picking wrong indexes with exporting models with bs>1 in decoding modules (YoloNAS, YoloNAS-Pose, YoloX, PPYolo-E)

 2) When using ONNX NonMaxSupression a max_predictions_per_image parameter was used incorrectly (Flat/Batch format predictions could contain more than a requested number of detections)
    @BloodAxe
    BloodAxe committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    27b17d5 View commit details
    Browse the repository at this point in the history

Commits on Oct 20, 2023

  1. Fixing NMS postprocessing to make it TRT compatible

    @BloodAxe
    BloodAxe committed Oct 20, 2023
    Configuration menu
    Copy the full SHA
    0567646 View commit details
    Browse the repository at this point in the history

  2. Update pose estimation support matrix for exported models

    @BloodAxe
    BloodAxe committed Oct 20, 2023
    Configuration menu
    Copy the full SHA
    d115f23 View commit details
    Browse the repository at this point in the history

  3. Update compatibility matrix for TRT NMS

    @BloodAxe
    BloodAxe committed Oct 20, 2023
    Configuration menu
    Copy the full SHA
    c155299 View commit details
    Browse the repository at this point in the history

Commits on Oct 21, 2023

  1. Improve ConvertTRTFormatToFlatTensor implementation

    @BloodAxe
    BloodAxe committed Oct 21, 2023
    Configuration menu
    Copy the full SHA
    4dd2cff View commit details
    Browse the repository at this point in the history

  2. Improve ConvertTRTFormatToFlatTensor implementation

    @BloodAxe
    BloodAxe committed Oct 21, 2023
    Configuration menu
    Copy the full SHA
    ad0e1c2 View commit details
    Browse the repository at this point in the history

  3. Improve ConvertTRTFormatToFlatTensor implementation

    @BloodAxe
    BloodAxe committed Oct 21, 2023
    Configuration menu
    Copy the full SHA
    21ac0bc View commit details
    Browse the repository at this point in the history

  4. Drop use_boolean_gather

    @BloodAxe
    BloodAxe committed Oct 21, 2023
    Configuration menu
    Copy the full SHA
    efe21a6 View commit details
    Browse the repository at this point in the history

  5. Improve test

    @BloodAxe
    BloodAxe committed Oct 21, 2023
    Configuration menu
    Copy the full SHA
    4db1a69 View commit details
    Browse the repository at this point in the history

  6. Update compatibility matrix

    @BloodAxe
    BloodAxe committed Oct 21, 2023
    Configuration menu
    Copy the full SHA
    0f788e0 View commit details
    Browse the repository at this point in the history

  7. Improve test

    @BloodAxe
    BloodAxe committed Oct 21, 2023
    Configuration menu
    Copy the full SHA
    4cc6a41 View commit details
    Browse the repository at this point in the history

  8. Improve test

    @BloodAxe
    BloodAxe committed Oct 21, 2023
    Configuration menu
    Copy the full SHA
    f754184 View commit details
    Browse the repository at this point in the history

Commits on Oct 23, 2023

  1. Merge branch 'master' into feature/SG-1195-fix-nms

    @BloodAxe
    BloodAxe committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    4b8559d View commit details
    Browse the repository at this point in the history

  2. Merge branch 'master' into feature/SG-1195-fix-nms

    @BloodAxe
    BloodAxe committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    fcef28c View commit details
    Browse the repository at this point in the history

  3. Update notebook export

    @BloodAxe
    BloodAxe committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    05d7913 View commit details
    Browse the repository at this point in the history

  4. Merge remote-tracking branch 'origin/feature/SG-1195-fix-nms' into fe… 

    …ature/SG-1195-fix-nms
    @BloodAxe
    BloodAxe committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    fffe13f View commit details
    Browse the repository at this point in the history

  5. Merge branch 'master' into feature/SG-1195-fix-nms

    @BloodAxe
    BloodAxe committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    3480373 View commit details
    Browse the repository at this point in the history

  6. Merge branch 'master' into feature/SG-1195-fix-nms

    @BloodAxe
    BloodAxe committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    cb82ab3 View commit details
    Browse the repository at this point in the history