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.

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

Segment TensorRT batch=3 fix #7952

Merged
merged 7 commits into from
Feb 1, 2024
Merged

Commits on Feb 1, 2024

  1. Update predict.py for segment

    When a pt model is exported to an engine with a fixed batchsize of 3 will this fail. Since len(np.array((3, 3, width, height))) is also true, so the code will continue like it is a pt model
    DieterTimmers committed Feb 1, 2024
    Configuration menu
    Copy the full SHA
    7b9550b View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    e503b34 View commit details
    Browse the repository at this point in the history
  3. Update predict.py

    Fixed the typo
    DieterTimmers committed Feb 1, 2024
    Configuration menu
    Copy the full SHA
    f4cd937 View commit details
    Browse the repository at this point in the history
  4. Update predict.py

    glenn-jocher committed Feb 1, 2024
    Configuration menu
    Copy the full SHA
    374c0d0 View commit details
    Browse the repository at this point in the history
  5. Configuration menu
    Copy the full SHA
    cce005b View commit details
    Browse the repository at this point in the history
  6. Update predict.py

    glenn-jocher committed Feb 1, 2024
    Configuration menu
    Copy the full SHA
    541e510 View commit details
    Browse the repository at this point in the history
  7. Configuration menu
    Copy the full SHA
    32a8979 View commit details
    Browse the repository at this point in the history