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

Horizon detection with YOLOv5's backbone #3

Merged
merged 66 commits into from
May 28, 2024
Merged

Horizon detection with YOLOv5's backbone #3

merged 66 commits into from
May 28, 2024

Commits on Oct 10, 2023

  1. added horizon utils

    @kevinconka
    kevinconka committed Oct 10, 2023
    Configuration menu
    Copy the full SHA
    ee72f4c View commit details
    Browse the repository at this point in the history

  2. Added dataloaders

    @kevinconka
    kevinconka committed Oct 10, 2023
    Configuration menu
    Copy the full SHA
    5f1432f View commit details
    Browse the repository at this point in the history

  3. rename to avoid confusion w/ yolov5/utils folder

    @franziska-denk
    franziska-denk committed Oct 10, 2023
    Configuration menu
    Copy the full SHA
    f430e3c View commit details
    Browse the repository at this point in the history

  4. add pitch and yaw targets

    @franziska-denk
    franziska-denk committed Oct 10, 2023
    Configuration menu
    Copy the full SHA
    28da609 View commit details
    Browse the repository at this point in the history

Commits on Oct 11, 2023

  1. HorizonDataset working w albumentations transform

    @kevinconka
    kevinconka committed Oct 11, 2023
    Configuration menu
    Copy the full SHA
    046025e View commit details
    Browse the repository at this point in the history

Commits on Oct 16, 2023

  1. Training loop working (needs refactoring)

    @kevinconka
    kevinconka committed Oct 16, 2023
    Configuration menu
    Copy the full SHA
    ea51021 View commit details
    Browse the repository at this point in the history

  2. Create README.md 

    Signed-off-by: Kevin Serrano <35779409+kevinconka@users.noreply.github.com>
    @kevinconka
    kevinconka authored Oct 16, 2023
    Configuration menu
    Copy the full SHA
    e8b349c View commit details
    Browse the repository at this point in the history

Commits on Oct 18, 2023

  1. refactored models.py

    @kevinconka
    kevinconka committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    c7829bd View commit details
    Browse the repository at this point in the history

  2. Added TODOs to README

    @kevinconka
    kevinconka committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    98c7d66 View commit details
    Browse the repository at this point in the history

  3. added train.py and transforms.py to cleanup nb

    @kevinconka
    kevinconka committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    d62c1bb View commit details
    Browse the repository at this point in the history

  4. Checked some TODOs from horizon/README.md

    @kevinconka
    kevinconka committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    1bc988c View commit details
    Browse the repository at this point in the history

  5. updated tests nb

    @kevinconka
    kevinconka committed Oct 18, 2023
    Configuration menu
    Copy the full SHA
    5747bf1 View commit details
    Browse the repository at this point in the history

Commits on Oct 19, 2023

  1. Save val/train loss plots as png images

    @kevinconka
    kevinconka committed Oct 19, 2023
    Configuration menu
    Copy the full SHA
    2e983ed View commit details
    Browse the repository at this point in the history

Commits on Oct 20, 2023

  1. Added loss weigths for horizon training

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

Commits on Oct 23, 2023

  1. Added more TODOs

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

Commits on Oct 25, 2023

  1. Added weather augmentations to rgb

    @kevinconka
    kevinconka committed Oct 25, 2023
    Configuration menu
    Copy the full SHA
    22e72e2 View commit details
    Browse the repository at this point in the history

Commits on Dec 24, 2023

  1. tensorrt engine working (WIP)

    @kevinconka
    kevinconka committed Dec 24, 2023
    Configuration menu
    Copy the full SHA
    6239c4f View commit details
    Browse the repository at this point in the history

  2. added patch for tensorRT v7

    @kevinconka
    kevinconka committed Dec 24, 2023
    Configuration menu
    Copy the full SHA
    67365a9 View commit details
    Browse the repository at this point in the history

Commits on Dec 26, 2023

  1. Update README.md 

    Signed-off-by: Kevin Serrano <35779409+kevinconka@users.noreply.github.com>
    @kevinconka
    kevinconka authored Dec 26, 2023
    Configuration menu
    Copy the full SHA
    fcba79a View commit details
    Browse the repository at this point in the history

Commits on Feb 5, 2024

  1. Update models.py 

    Signed-off-by: Kevin Serrano <35779409+kevinconka@users.noreply.github.com>
    @kevinconka
    kevinconka authored Feb 5, 2024
    Configuration menu
    Copy the full SHA
    9d8fe07 View commit details
    Browse the repository at this point in the history

Commits on Feb 11, 2024

  1. Merge branch 'sea.ai' into feature/horizon-det

    @kevinconka
    kevinconka committed Feb 11, 2024
    Configuration menu
    Copy the full SHA
    8145e6d View commit details
    Browse the repository at this point in the history

Commits on Mar 1, 2024

  1. Added batch support for yolotrt

    @kevinconka
    kevinconka committed Mar 1, 2024
    Configuration menu
    Copy the full SHA
    574c2c5 View commit details
    Browse the repository at this point in the history

Commits on Mar 2, 2024

  1. folder refactoring

    @kevinconka
    kevinconka committed Mar 2, 2024
    Configuration menu
    Copy the full SHA
    abd6971 View commit details
    Browse the repository at this point in the history

  2. more refactoring... added "hybrids" folder

    @kevinconka
    kevinconka committed Mar 2, 2024
    Configuration menu
    Copy the full SHA
    c3b7c22 View commit details
    Browse the repository at this point in the history

Commits on Mar 4, 2024

  1. Update yolotrt.py 

    Signed-off-by: Kevin Serrano <35779409+kevinconka@users.noreply.github.com>
    @kevinconka
    kevinconka authored Mar 4, 2024
    Configuration menu
    Copy the full SHA
    1580c4a View commit details
    Browse the repository at this point in the history

  2. working nb for ahoy-IR.engine

    @kevinconka
    kevinconka committed Mar 4, 2024
    Configuration menu
    Copy the full SHA
    7b59839 View commit details
    Browse the repository at this point in the history

Commits on Mar 10, 2024

  1. AHOYv5 and DANv5 refactored

    @kevinconka
    kevinconka committed Mar 10, 2024
    Configuration menu
    Copy the full SHA
    4219024 View commit details
    Browse the repository at this point in the history

Commits on Mar 11, 2024

  1. typing compatibility with python3.8

    @kevinconka
    kevinconka committed Mar 11, 2024
    Configuration menu
    Copy the full SHA
    713767e View commit details
    Browse the repository at this point in the history

  2. get images from url and use softmax for horizon postprocessing in dan… 

    …_tests
    @kevinconka
    kevinconka committed Mar 11, 2024
    Configuration menu
    Copy the full SHA
    6647f33 View commit details
    Browse the repository at this point in the history

Commits on Mar 12, 2024

  1. made output_modes compatible with batched inference for ahoy

    @kevinconka
    kevinconka committed Mar 12, 2024
    Configuration menu
    Copy the full SHA
    d784fee View commit details
    Browse the repository at this point in the history

  2. tweakes plus extended export function for AHOY and DAN models

    @kevinconka
    kevinconka committed Mar 12, 2024
    Configuration menu
    Copy the full SHA
    585e0bf View commit details
    Browse the repository at this point in the history

  3. Add benchmarking script for AHOY and DAN models

    @kevinconka
    kevinconka committed Mar 12, 2024
    Configuration menu
    Copy the full SHA
    b346bca View commit details
    Browse the repository at this point in the history

  4. fixed misspelling

    @kevinconka
    kevinconka committed Mar 12, 2024
    Configuration menu
    Copy the full SHA
    5d538d8 View commit details
    Browse the repository at this point in the history

Commits on Mar 13, 2024

  1. new preprocessing function

    @kevinconka
    kevinconka committed Mar 13, 2024
    Configuration menu
    Copy the full SHA
    708e062 View commit details
    Browse the repository at this point in the history

Commits on Mar 15, 2024

  1. faster preprocessing

    @kevinconka
    kevinconka committed Mar 15, 2024
    Configuration menu
    Copy the full SHA
    28a0237 View commit details
    Browse the repository at this point in the history

Commits on Mar 16, 2024

  1. fast preprocess

    @kevinconka
    kevinconka committed Mar 16, 2024
    Configuration menu
    Copy the full SHA
    5e453b5 View commit details
    Browse the repository at this point in the history

Commits on Mar 23, 2024

  1. added class mapping to dan

    @kevinconka
    kevinconka committed Mar 23, 2024
    Configuration menu
    Copy the full SHA
    cd87cdb View commit details
    Browse the repository at this point in the history

  2. refactored dan cls_maps

    @kevinconka
    kevinconka committed Mar 23, 2024
    Configuration menu
    Copy the full SHA
    5ceada5 View commit details
    Browse the repository at this point in the history

Commits on Mar 24, 2024

  1. refactored trt inference code after adding it to Core-Backend

    @kevinconka
    kevinconka committed Mar 24, 2024
    Configuration menu
    Copy the full SHA
    039727a View commit details
    Browse the repository at this point in the history

Commits on Mar 25, 2024

  1. fixed bug in horizon output decoding when lr padding was added during… 

    … inference
    @kevinconka
    kevinconka committed Mar 25, 2024
    Configuration menu
    Copy the full SHA
    abf3df6 View commit details
    Browse the repository at this point in the history

  2. added small comment

    @kevinconka
    kevinconka committed Mar 25, 2024
    Configuration menu
    Copy the full SHA
    0d17d9e View commit details
    Browse the repository at this point in the history

Commits on Mar 26, 2024

  1. add export hooks to get fp32 to fp16 done in GPU

    @kevinconka
    kevinconka committed Mar 26, 2024
    Configuration menu
    Copy the full SHA
    c7d2178 View commit details
    Browse the repository at this point in the history

  2. register hooks before dry runs when exporting hybrids

    @kevinconka
    kevinconka committed Mar 26, 2024
    Configuration menu
    Copy the full SHA
    ceaf2e3 View commit details
    Browse the repository at this point in the history

  3. added preprocessing operations to the model via hooks before trt export

    @kevinconka
    kevinconka committed Mar 26, 2024
    Configuration menu
    Copy the full SHA
    9c9539c View commit details
    Browse the repository at this point in the history

  4. added hook to export process

    @kevinconka
    kevinconka committed Mar 26, 2024
    Configuration menu
    Copy the full SHA
    a224373 View commit details
    Browse the repository at this point in the history

  5. make dummy input int8 for trt export

    @kevinconka
    kevinconka committed Mar 26, 2024
    Configuration menu
    Copy the full SHA
    5cb1e3c View commit details
    Browse the repository at this point in the history

Commits on Mar 27, 2024

  1. added softmax to postprocessing hook to add it to the model

    @kevinconka
    kevinconka committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    79ccc2e View commit details
    Browse the repository at this point in the history

  2. remove softmax from theta, offset postprocessing

    @kevinconka
    kevinconka committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    477f5e7 View commit details
    Browse the repository at this point in the history

Commits on Apr 24, 2024

  1. Refactor model variable names in export.py and export_hybrid.py

    @kevinconka
    kevinconka committed Apr 24, 2024
    Configuration menu
    Copy the full SHA
    2fddcb4 View commit details
    Browse the repository at this point in the history

Commits on Apr 28, 2024

  1. ResizeIfNeeded augmentation and plot MSEs during training

    @kevinconka
    kevinconka committed Apr 28, 2024
    Configuration menu
    Copy the full SHA
    1537ae3 View commit details
    Browse the repository at this point in the history

  2. Update num_workers in dataloaders.py

    @kevinconka
    kevinconka committed Apr 28, 2024
    Configuration menu
    Copy the full SHA
    6ccb7e1 View commit details
    Browse the repository at this point in the history

  3. Refactor horizon.py and custom.py for better readability and maintain… 

    …ability.
    @kevinconka
    kevinconka committed Apr 28, 2024
    Configuration menu
    Copy the full SHA
    9ac2803 View commit details
    Browse the repository at this point in the history

  4. Add best_mse.pt checkpoint saving during training

    @kevinconka
    kevinconka committed Apr 28, 2024
    Configuration menu
    Copy the full SHA
    c708bc5 View commit details
    Browse the repository at this point in the history

  5. Fix data type conversion issue in custom.py

    @kevinconka
    kevinconka committed Apr 28, 2024
    Configuration menu
    Copy the full SHA
    2a23bc7 View commit details
    Browse the repository at this point in the history

  6. Fix data type conversion issue in train.py

    @kevinconka
    kevinconka committed Apr 28, 2024
    Configuration menu
    Copy the full SHA
    8e80a6d View commit details
    Browse the repository at this point in the history

Commits on Apr 29, 2024

  1. wandb integration working

    @kevinconka
    kevinconka committed Apr 29, 2024
    Configuration menu
    Copy the full SHA
    47e3d71 View commit details
    Browse the repository at this point in the history

  2. cleaned nb

    @kevinconka
    kevinconka committed Apr 29, 2024
    Configuration menu
    Copy the full SHA
    9fa1b01 View commit details
    Browse the repository at this point in the history

  3. scale pitch for gt and preds when processing 320x256 imgs

    @kevinconka
    kevinconka committed Apr 29, 2024
    Configuration menu
    Copy the full SHA
    ebbd17c View commit details
    Browse the repository at this point in the history

  4. fix msu_sum and better cli logging

    @kevinconka
    kevinconka committed Apr 29, 2024
    Configuration menu
    Copy the full SHA
    e205d58 View commit details
    Browse the repository at this point in the history

  5. fixed mask visualisation

    @kevinconka
    kevinconka committed Apr 29, 2024
    Configuration menu
    Copy the full SHA
    ef17cd0 View commit details
    Browse the repository at this point in the history

  6. cast to int32 for cv2.line

    @kevinconka
    kevinconka committed Apr 29, 2024
    Configuration menu
    Copy the full SHA
    a8c4304 View commit details
    Browse the repository at this point in the history

  7. Fix formatting issue in cli logging

    @kevinconka
    kevinconka committed Apr 29, 2024
    Configuration menu
    Copy the full SHA
    1009c00 View commit details
    Browse the repository at this point in the history

Commits on May 28, 2024

  1. minor notebook update

    @kevinconka
    kevinconka committed May 28, 2024
    Configuration menu
    Copy the full SHA
    53c98c5 View commit details
    Browse the repository at this point in the history

  2. Merge remote-tracking branch 'origin/sea.ai' into feature/horizon-det

    @kevinconka
    kevinconka committed May 28, 2024
    Configuration menu
    Copy the full SHA
    14855bb View commit details
    Browse the repository at this point in the history

  3. Function docstrings added

    @kevinconka
    kevinconka committed May 28, 2024
    Configuration menu
    Copy the full SHA
    550902a View commit details
    Browse the repository at this point in the history

  4. Auto-format by https://ultralytics.com/actions

    @UltralyticsAssistant
    UltralyticsAssistant committed May 28, 2024
    Configuration menu
    Copy the full SHA
    4cf1773 View commit details
    Browse the repository at this point in the history