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

PyTorch Object Detector Training Implementation and Prediction Update #2067

Merged
merged 16 commits into from
Mar 15, 2023
Merged

PyTorch Object Detector Training Implementation and Prediction Update #2067

merged 16 commits into from
Mar 15, 2023

Commits on Mar 12, 2023

  1. implement model training for pytorch faster rcnn 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 12, 2023
    Configuration menu
    Copy the full SHA
    0d70547 View commit details
    Browse the repository at this point in the history

  2. use batches for pytorch faster rcnn predictions 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 12, 2023
    Configuration menu
    Copy the full SHA
    5287f9e View commit details
    Browse the repository at this point in the history

  3. implement model training for pytorch yolo 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 12, 2023
    Configuration menu
    Copy the full SHA
    3cc3083 View commit details
    Browse the repository at this point in the history

  4. added pytorch yolo test for fit 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 12, 2023
    Configuration menu
    Copy the full SHA
    8ce4eca View commit details
    Browse the repository at this point in the history

  5. added missing pytorch yolo estimator params 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 12, 2023
    Configuration menu
    Copy the full SHA
    3b95fb0 View commit details
    Browse the repository at this point in the history

  6. update predict batching 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 12, 2023
    Configuration menu
    Copy the full SHA
    3fb0f52 View commit details
    Browse the repository at this point in the history

  7. update BadDet notebook with model fitting 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 12, 2023
    Configuration menu
    Copy the full SHA
    f687f14 View commit details
    Browse the repository at this point in the history

Commits on Mar 13, 2023

  1. rewrite pytorch object detector tests in pytest 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    86312e0 View commit details
    Browse the repository at this point in the history

  2. cleanup pytorch yolo and object detector tests 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    a4d0f87 View commit details
    Browse the repository at this point in the history

  3. convert pytorch faster rcnn test cases to pytest 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    aa88c72 View commit details
    Browse the repository at this point in the history

  4. fix style checks 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    0ba2e62 View commit details
    Browse the repository at this point in the history

  5. address review comments 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    9109b27 View commit details
    Browse the repository at this point in the history

  6. fix style checks 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    98f6d05 View commit details
    Browse the repository at this point in the history

  7. address remainder of review comments 

    Signed-off-by: Farhan Ahmed <Farhan.Ahmed@ibm.com>
    @f4str
    f4str committed Mar 13, 2023
    Configuration menu
    Copy the full SHA
    d4e326c View commit details
    Browse the repository at this point in the history

Commits on Mar 14, 2023

  1. Merge branch 'dev_1.14.0' into pytorch-object-detector-training

    @beat-buesser
    beat-buesser committed Mar 14, 2023
    Configuration menu
    Copy the full SHA
    8a6ecf8 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'dev_1.14.0' into pytorch-object-detector-training

    @beat-buesser
    beat-buesser committed Mar 14, 2023
    Configuration menu
    Copy the full SHA
    f774359 View commit details
    Browse the repository at this point in the history