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 YOLO object detection model #552

Merged
merged 86 commits into from
Sep 10, 2021
Merged

Add YOLO object detection model #552

merged 86 commits into from
Sep 10, 2021

Commits on Feb 2, 2021

  1. Add YOLO object detection model

    @senarvi
    senarvi committed Feb 2, 2021
    Configuration menu
    Copy the full SHA
    461de96 View commit details
    Browse the repository at this point in the history

Commits on Feb 3, 2021

  1. Readability improvements

    @senarvi
    senarvi committed Feb 3, 2021
    Configuration menu
    Copy the full SHA
    2b9b073 View commit details
    Browse the repository at this point in the history

  2. Documentation improvements

    @senarvi
    senarvi committed Feb 3, 2021
    Configuration menu
    Copy the full SHA
    cc42540 View commit details
    Browse the repository at this point in the history

  3. Fixed style issues.

    @senarvi
    senarvi committed Feb 3, 2021
    Configuration menu
    Copy the full SHA
    876da0d View commit details
    Browse the repository at this point in the history

Commits on Feb 5, 2021

  1. Refactoring

    @senarvi
    senarvi committed Feb 5, 2021
    Configuration menu
    Copy the full SHA
    f99930e View commit details
    Browse the repository at this point in the history

Commits on Feb 8, 2021

  1. Refactoring

    @senarvi
    senarvi committed Feb 8, 2021
    Configuration menu
    Copy the full SHA
    4415d41 View commit details
    Browse the repository at this point in the history

  2. Refactoring

    @senarvi
    senarvi committed Feb 8, 2021
    Configuration menu
    Copy the full SHA
    7356fe0 View commit details
    Browse the repository at this point in the history

  3. Fixed YOLO test.

    @senarvi
    senarvi committed Feb 8, 2021
    Configuration menu
    Copy the full SHA
    39eb80d View commit details
    Browse the repository at this point in the history

Commits on Feb 9, 2021

  1. Fixedd style issues

    @senarvi
    senarvi committed Feb 9, 2021
    Configuration menu
    Copy the full SHA
    291f4be View commit details
    Browse the repository at this point in the history

  2. Comply to isort rules.

    @senarvi
    senarvi committed Feb 9, 2021
    Configuration menu
    Copy the full SHA
    8db7947 View commit details
    Browse the repository at this point in the history

  3. Reading Darknet weights works also with truncated files.

    @senarvi
    senarvi committed Feb 9, 2021
    Configuration menu
    Copy the full SHA
    2831755 View commit details
    Browse the repository at this point in the history

  4. Fixed code formatting.

    @senarvi
    senarvi committed Feb 9, 2021
    Configuration menu
    Copy the full SHA
    eb26eba View commit details
    Browse the repository at this point in the history

  5. Trying to fix Python 3.6 import problem.

    @senarvi
    senarvi committed Feb 9, 2021
    Configuration menu
    Copy the full SHA
    9c155a9 View commit details
    Browse the repository at this point in the history

  6. Fixed Python 3.6 import error.

    @senarvi
    senarvi committed Feb 9, 2021
    Configuration menu
    Copy the full SHA
    efeb1c8 View commit details
    Browse the repository at this point in the history

  7. Added YOLO to CHANGELOG.

    @senarvi
    senarvi committed Feb 9, 2021
    Configuration menu
    Copy the full SHA
    1a1ecd3 View commit details
    Browse the repository at this point in the history

  8. Use torch.min() instead of torch.minimum() to avoid error with older … 

    …PyTorch versions.
    @senarvi
    senarvi committed Feb 9, 2021
    Configuration menu
    Copy the full SHA
    26ff979 View commit details
    Browse the repository at this point in the history

Commits on Feb 12, 2021

  1. Generalized interface for custom losses 

    * IoU loss functions take image space coordinates as input.
    @senarvi
    senarvi committed Feb 12, 2021
    Configuration menu
    Copy the full SHA
    3e9bdde View commit details
    Browse the repository at this point in the history

  2. box_area() implementation copied from torchvision 

    * box_area() is not found from the version of torchvision used by the test suite.
    @senarvi
    senarvi committed Feb 12, 2021
    Configuration menu
    Copy the full SHA
    c348619 View commit details
    Browse the repository at this point in the history

  3. Confirm to yapf formatter rules.

    @senarvi
    senarvi committed Feb 12, 2021
    Configuration menu
    Copy the full SHA
    c2d7907 View commit details
    Browse the repository at this point in the history

Commits on Feb 15, 2021

  1. Removed the unnecessary linter instructions. 

    Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
    @senarvi @akihironitta
    senarvi and akihironitta authored Feb 15, 2021
    Configuration menu
    Copy the full SHA
    3d7f440 View commit details
    Browse the repository at this point in the history

  2. IoU losses use torchvision 

    * IoU losses take the diagnoal of torchvision iou ops instead of implementing their own elementwise ops.
* Quotes replaced with double quotes in docstrings.
* Import _TORCHVISION_AVAILABLE from pl_bolts.
    @senarvi
    senarvi committed Feb 15, 2021
    Configuration menu
    Copy the full SHA
    eb6be46 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'yolo' of github.com:groke-technologies/pytorch-lightnin… 

    …g-bolts into yolo
    @senarvi
    senarvi committed Feb 15, 2021
    Configuration menu
    Copy the full SHA
    3d940f2 View commit details
    Browse the repository at this point in the history

  4. Improved strange yapf formatting 

    Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
    @senarvi @akihironitta
    senarvi and akihironitta authored Feb 15, 2021
    Configuration menu
    Copy the full SHA
    cf7420c View commit details
    Browse the repository at this point in the history

  5. Refactoring 

    * YOLO written with all caps in class names
* Generic way to specify optimizer and LR scheduler
    @senarvi
    senarvi committed Feb 15, 2021
    Configuration menu
    Copy the full SHA
    6c90cd4 View commit details
    Browse the repository at this point in the history

  6. Merge branch 'yolo' of github.com:groke-technologies/pytorch-lightnin… 

    …g-bolts into yolo
    @senarvi
    senarvi committed Feb 15, 2021
    Configuration menu
    Copy the full SHA
    7aeea63 View commit details
    Browse the repository at this point in the history

  7. get_deprecated_arg_names() is not needed anymore.

    @senarvi
    senarvi committed Feb 15, 2021
    Configuration menu
    Copy the full SHA
    60fda75 View commit details
    Browse the repository at this point in the history

  8. Fixed yapf formatting. 

    Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
    @senarvi @akihironitta
    senarvi and akihironitta authored Feb 15, 2021
    Configuration menu
    Copy the full SHA
    b2e3e84 View commit details
    Browse the repository at this point in the history

  9. Fixed formatting. 

    Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
    @senarvi @akihironitta
    senarvi and akihironitta authored Feb 15, 2021
    Configuration menu
    Copy the full SHA
    940947f View commit details
    Browse the repository at this point in the history

  10. Removed unused imports.

    @senarvi
    senarvi committed Feb 15, 2021
    Configuration menu
    Copy the full SHA
    6e3d5bf View commit details
    Browse the repository at this point in the history

  11. Merge branch 'yolo' of github.com:groke-technologies/pytorch-lightnin… 

    …g-bolts into yolo
    @senarvi
    senarvi committed Feb 15, 2021
    Configuration menu
    Copy the full SHA
    a5bed26 View commit details
    Browse the repository at this point in the history

Commits on Feb 16, 2021

  1. Fixed some type hints.

    @senarvi
    senarvi committed Feb 16, 2021
    Configuration menu
    Copy the full SHA
    b2ea497 View commit details
    Browse the repository at this point in the history

  2. Sorted imports.

    @senarvi
    senarvi committed Feb 16, 2021
    Configuration menu
    Copy the full SHA
    6d8fa7d View commit details
    Browse the repository at this point in the history

Commits on Feb 24, 2021

  1. Possible to limit the number of predictions per image

    @senarvi
    senarvi committed Feb 24, 2021
    Configuration menu
    Copy the full SHA
    e68df7a View commit details
    Browse the repository at this point in the history

  2. None instead of an empty list as default argument

    @senarvi
    senarvi committed Feb 24, 2021
    Configuration menu
    Copy the full SHA
    f895530 View commit details
    Browse the repository at this point in the history

  3. Fixed capitalization of YOLO class.

    @senarvi
    senarvi committed Feb 24, 2021
    Configuration menu
    Copy the full SHA
    58f1456 View commit details
    Browse the repository at this point in the history

Commits on Mar 4, 2021

  1. Merge branch 'origin/master' into yolo

    @senarvi
    senarvi committed Mar 4, 2021
    Configuration menu
    Copy the full SHA
    19b9df7 View commit details
    Browse the repository at this point in the history

Commits on Mar 8, 2021

  1. No need to check for NaN values as Trainer has terminate_on_nan argum… 

    …ent.
    @senarvi
    senarvi committed Mar 8, 2021
    Configuration menu
    Copy the full SHA
    4e6d4cf View commit details
    Browse the repository at this point in the history

  2. YOLO test configuration moved to tests/data/yolo.cfg

    @senarvi
    senarvi committed Mar 8, 2021
    Configuration menu
    Copy the full SHA
    af3e0e6 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'origin/master' into yolo

    @senarvi
    senarvi committed Mar 8, 2021
    Configuration menu
    Copy the full SHA
    c4ae5ec View commit details
    Browse the repository at this point in the history

  4. Use Optional[] as the default value for transforms is now None

    @senarvi
    senarvi committed Mar 8, 2021
    Configuration menu
    Copy the full SHA
    4012247 View commit details
    Browse the repository at this point in the history

Commits on Mar 23, 2021

  1. Refactoring and documentation improvements 

    * Synchronize validation and test step logging calls
* Log losses to progress bar
    @senarvi
    senarvi committed Mar 23, 2021
    Configuration menu
    Copy the full SHA
    c8b76a5 View commit details
    Browse the repository at this point in the history

Commits on Mar 24, 2021

  1. Fixed documentation formatting

    @senarvi
    senarvi committed Mar 24, 2021
    Configuration menu
    Copy the full SHA
    71a4c3c View commit details
    Browse the repository at this point in the history

  2. Merge branch 'origin/master' into yolo

    @senarvi
    senarvi committed Mar 24, 2021
    Configuration menu
    Copy the full SHA
    da4eace View commit details
    Browse the repository at this point in the history

Commits on Mar 31, 2021

  1. Coordinate predictions are in image scale

    @senarvi
    senarvi committed Mar 31, 2021
    Configuration menu
    Copy the full SHA
    70f14b0 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'origin/master' into yolo

    @senarvi
    senarvi committed Mar 31, 2021
    Configuration menu
    Copy the full SHA
    3c1a0fb View commit details
    Browse the repository at this point in the history

Commits on Apr 1, 2021

  1. Use default dtype for torch.arange() to fix export to TensorRT

    @senarvi
    senarvi committed Apr 1, 2021
    Configuration menu
    Copy the full SHA
    9587b46 View commit details
    Browse the repository at this point in the history

Commits on Apr 10, 2021

  1. Network input size can differ from the image size specified in the co… 

    …nfiguration

* Image size is given to detection layer forward() instead of the constructor to allow variable image sizes.
* Use default data type for torch.arange() to fix export to TensorRT.
    @senarvi
    senarvi committed Apr 10, 2021
    Configuration menu
    Copy the full SHA
    2b6c552 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'yolo' of github.com:groke-technologies/pytorch-lightnin… 

    …g-bolts into yolo
    @senarvi
    senarvi committed Apr 10, 2021
    Configuration menu
    Copy the full SHA
    e97b198 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'origin/master' into yolo

    @senarvi
    senarvi committed Apr 10, 2021
    Configuration menu
    Copy the full SHA
    dbe1d59 View commit details
    Browse the repository at this point in the history

Commits on May 3, 2021

  1. Merge branch 'master' into yolo

    @senarvi
    senarvi committed May 3, 2021
    Configuration menu
    Copy the full SHA
    b66dbd3 View commit details
    Browse the repository at this point in the history

  2. Use torch.true_divide() instead of /

    @senarvi
    senarvi committed May 3, 2021
    Configuration menu
    Copy the full SHA
    004d1ce View commit details
    Browse the repository at this point in the history

Commits on May 5, 2021

  1. Use torch.true_divide() instead of /

    @senarvi
    senarvi committed May 5, 2021
    Configuration menu
    Copy the full SHA
    ad1e48e View commit details
    Browse the repository at this point in the history

Commits on May 13, 2021

  1. Merge branch 'master' into yolo

    @senarvi
    senarvi committed May 13, 2021
    Configuration menu
    Copy the full SHA
    a254517 View commit details
    Browse the repository at this point in the history

Commits on Jun 1, 2021

  1. Merge branch 'origin/master' into yolo

    @senarvi
    senarvi committed Jun 1, 2021
    Configuration menu
    Copy the full SHA
    1cde4f8 View commit details
    Browse the repository at this point in the history

Commits on Jun 17, 2021

  1. Merge branch 'master' into yolo

    @senarvi
    senarvi committed Jun 17, 2021
    Configuration menu
    Copy the full SHA
    e92c405 View commit details
    Browse the repository at this point in the history

Commits on Jun 23, 2021

  1. Loss is normalized by batch size only once

    @senarvi
    senarvi committed Jun 23, 2021
    Configuration menu
    Copy the full SHA
    c237b37 View commit details
    Browse the repository at this point in the history

  2. Fixed division by zero when there are no targets in a batch

    @senarvi
    senarvi committed Jun 23, 2021
    Configuration menu
    Copy the full SHA
    9b010de View commit details
    Browse the repository at this point in the history

  3. Merge branch 'master' into yolo

    @senarvi
    senarvi committed Jun 23, 2021
    Configuration menu
    Copy the full SHA
    dc7ae4c View commit details
    Browse the repository at this point in the history

Commits on Jun 24, 2021

  1. Always return all losses to avoid deadlock with DDP when there are no… 

    … targets
    @senarvi
    senarvi committed Jun 24, 2021
    Configuration menu
    Copy the full SHA
    f15282d View commit details
    Browse the repository at this point in the history

  2. Merge branch 'master' into yolo

    @senarvi
    senarvi committed Jun 24, 2021
    Configuration menu
    Copy the full SHA
    de52b75 View commit details
    Browse the repository at this point in the history

Commits on Jul 1, 2021

  1. Hit rates are always logged so don't prefix the names

    @senarvi
    senarvi committed Jul 1, 2021
    Configuration menu
    Copy the full SHA
    f6d3476 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'master' into yolo

    @senarvi
    senarvi committed Jul 1, 2021
    Configuration menu
    Copy the full SHA
    8e12359 View commit details
    Browse the repository at this point in the history

Commits on Jul 31, 2021

  1. Fixed training loss 

    * The vector of overlap losses was accidentally transformed to a square matrix.
* Some versions of Lightning don't work correctly when logging losses with sync_dist=True.
    @senarvi
    senarvi committed Jul 31, 2021
    Configuration menu
    Copy the full SHA
    86a6b66 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'origin/master' into yolo

    @senarvi
    senarvi committed Jul 31, 2021
    Configuration menu
    Copy the full SHA
    7fd38ca View commit details
    Browse the repository at this point in the history

Commits on Aug 4, 2021

  1. Truncate nms() inputs to avoid it crashing when too many boxes are de… 

    …tected
    @senarvi
    senarvi committed Aug 4, 2021
    Configuration menu
    Copy the full SHA
    3286533 View commit details
    Browse the repository at this point in the history

Commits on Aug 11, 2021

  1. Use sum() instead of count_nonzero() as it's available already before… 

    … PyTorch 1.7
    @senarvi
    senarvi committed Aug 11, 2021
    Configuration menu
    Copy the full SHA
    bb92076 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'master' into yolo

    @senarvi
    senarvi committed Aug 11, 2021
    Configuration menu
    Copy the full SHA
    7d08350 View commit details
    Browse the repository at this point in the history

Commits on Aug 17, 2021

  1. Squared error loss takes the sum over the predicted attributes

    @senarvi
    senarvi committed Aug 17, 2021
    Configuration menu
    Copy the full SHA
    b896112 View commit details
    Browse the repository at this point in the history

  2. Swish and logistic activation functions

    @senarvi
    senarvi committed Aug 17, 2021
    Configuration menu
    Copy the full SHA
    55a1180 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'master' into yolo

    @senarvi
    senarvi committed Aug 17, 2021
    Configuration menu
    Copy the full SHA
    6fb82c1 View commit details
    Browse the repository at this point in the history

Commits on Aug 18, 2021

  1. Added a comment

    @senarvi
    senarvi committed Aug 18, 2021
    Configuration menu
    Copy the full SHA
    7857bea View commit details
    Browse the repository at this point in the history

  2. Fixed code formatting

    @senarvi
    senarvi committed Aug 18, 2021
    Configuration menu
    Copy the full SHA
    0804699 View commit details
    Browse the repository at this point in the history

Commits on Aug 19, 2021

  1. Ran docformatter with correct config

    @senarvi
    senarvi committed Aug 19, 2021
    Configuration menu
    Copy the full SHA
    7b32d64 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'master' into yolo

    @senarvi
    senarvi committed Aug 19, 2021
    Configuration menu
    Copy the full SHA
    7b316b8 View commit details
    Browse the repository at this point in the history

  3. Ran pyupgrade

    @senarvi
    senarvi committed Aug 19, 2021
    Configuration menu
    Copy the full SHA
    191e865 View commit details
    Browse the repository at this point in the history

  4. Reformatted

    @senarvi
    senarvi committed Aug 19, 2021
    Configuration menu
    Copy the full SHA
    66c8111 View commit details
    Browse the repository at this point in the history

Commits on Aug 30, 2021

  1. VOCDetectionDataModule constructor takes batch size and the transform… 

    …s take image and target
    @senarvi
    senarvi committed Aug 30, 2021
    Configuration menu
    Copy the full SHA
    572ca4f View commit details
    Browse the repository at this point in the history

  2. Merge branch 'master' into yolo

    @senarvi
    senarvi committed Aug 30, 2021
    Configuration menu
    Copy the full SHA
    198dd32 View commit details
    Browse the repository at this point in the history

  3. Use true_divide() for integer division

    @senarvi
    senarvi committed Aug 30, 2021
    Configuration menu
    Copy the full SHA
    db278be View commit details
    Browse the repository at this point in the history

  4. Fixed doc and package build without Torchvision

    @senarvi
    senarvi committed Aug 30, 2021
    Configuration menu
    Copy the full SHA
    4731b24 View commit details
    Browse the repository at this point in the history

Commits on Aug 31, 2021

  1. Merge branch 'master' into yolo

    @senarvi
    senarvi committed Aug 31, 2021
    Configuration menu
    Copy the full SHA
    f63df0b View commit details
    Browse the repository at this point in the history

Commits on Sep 8, 2021

  1. Merge branch 'master' into yolo

    @senarvi
    senarvi committed Sep 8, 2021
    Configuration menu
    Copy the full SHA
    3b70b77 View commit details
    Browse the repository at this point in the history

Commits on Sep 10, 2021

  1. YOLO moved to unreleased

    @senarvi
    senarvi committed Sep 10, 2021
    Configuration menu
    Copy the full SHA
    022192a View commit details
    Browse the repository at this point in the history

  2. Merge branch 'master' into yolo

    @senarvi
    senarvi committed Sep 10, 2021
    Configuration menu
    Copy the full SHA
    a8e1ce3 View commit details
    Browse the repository at this point in the history

  3. Code formatting 

    Co-authored-by: Aki Nitta <nitta@akihironitta.com>
    @senarvi @akihironitta
    senarvi and akihironitta authored Sep 10, 2021
    Configuration menu
    Copy the full SHA
    bfc774c View commit details
    Browse the repository at this point in the history

  4. Merge branch 'yolo' of github.com:groke-technologies/pytorch-lightnin… 

    …g-bolts into yolo
    @senarvi
    senarvi committed Sep 10, 2021
    Configuration menu
    Copy the full SHA
    8d4e43e View commit details
    Browse the repository at this point in the history