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

Add TensorRT YOLO implementation #17

Draft
wants to merge 16 commits into
base: master
Choose a base branch
from
Draft

Commits on Mar 3, 2020

  1. added yolov3 submodule

    JHLee0513 committed Mar 3, 2020
    Configuration menu
    Copy the full SHA
    f920ed2 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    7027ef2 View commit details
    Browse the repository at this point in the history
  3. Make uw_detection a catkin package

    Wrap yolov3 submodule in a package, have the package use catkin python
    setup to provide the module to scripts and other catkin packages
    Paste camera demo with imports corrected
    nickswalker committed Mar 3, 2020
    Configuration menu
    Copy the full SHA
    0027d8b View commit details
    Browse the repository at this point in the history

Commits on Mar 14, 2020

  1. Configuration menu
    Copy the full SHA
    c39611b View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    614e03d View commit details
    Browse the repository at this point in the history

Commits on May 9, 2020

  1. Configuration menu
    Copy the full SHA
    cf3c942 View commit details
    Browse the repository at this point in the history

Commits on May 29, 2020

  1. Pull submodules in CI

    nickswalker committed May 29, 2020
    Configuration menu
    Copy the full SHA
    6392111 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    fd3b154 View commit details
    Browse the repository at this point in the history

Commits on Jun 24, 2020

  1. Trying to get a script working that:

    1. subscribes to image from sim
    2. predicts object bounding boxes
    3. publishes prediction
    Brian Lee committed Jun 24, 2020
    Configuration menu
    Copy the full SHA
    a33a1fc View commit details
    Browse the repository at this point in the history

Commits on Jul 22, 2020

  1. Add hacked together YOLOv3 TensorRT implementation

    Includes a copy of Xinyu Wang's TensorRT implementation of the Ultralytics YOLOv3 implementation
    Some headers headers shuffled out into the include directory so things build in a ROS environment
    A bit of CMake setup to make sure things only attempt to build when there's CUDA on the machine
    nickswalker committed Jul 22, 2020
    Configuration menu
    Copy the full SHA
    764063a View commit details
    Browse the repository at this point in the history

Commits on Aug 2, 2020

  1. Copy in and patch plain YOLOv3

    Change demo node to use plain model
    Add readme with usage notes
    nickswalker committed Aug 2, 2020
    Configuration menu
    Copy the full SHA
    77a7f2c View commit details
    Browse the repository at this point in the history
  2. Fix formatting

    Not touching the copy pasted code to make future diffing easier
    Add clang files
    nickswalker committed Aug 2, 2020
    Configuration menu
    Copy the full SHA
    16507ad View commit details
    Browse the repository at this point in the history

Commits on Aug 14, 2020

  1. Remove YOLOv3 submodule

    nickswalker committed Aug 14, 2020
    Configuration menu
    Copy the full SHA
    cf85164 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    5fc8ef5 View commit details
    Browse the repository at this point in the history
  3. Linter updates for YOLO

    Fix whitespace, disable roslint
    nickswalker committed Aug 14, 2020
    Configuration menu
    Copy the full SHA
    d88e143 View commit details
    Browse the repository at this point in the history

Commits on Sep 1, 2020

  1. Make class num more configurable

    Use TensorRTs plugin options struct to pass in the number. Also copied
    over some boilerplate for making other parameters configurable this way but
    didn't hook them up fully
    Tweak plugin registration point to match an update in upstream
    nickswalker committed Sep 1, 2020
    Configuration menu
    Copy the full SHA
    69839bd View commit details
    Browse the repository at this point in the history