Code

Add a tensorrt backend#33

Code

Draft

BestDriverCN wants to merge 22 commits intomasteropencv/open_vision_capsules:masterfrom tensorrt_backendopencv/open_vision_capsules:tensorrt_backendCopy head branch name to clipboard

+251-6

Commits

Commits on Apr 20, 2021

• add a base tensorrt backend

  

  Zhao

  committed
  3059a7e

Commits on Apr 22, 2021

• ravel before concatenate to improve performance

  

  Zhao

  committed
  17d6c48

Commits on Apr 27, 2021

• don't resize if the size of image remains the same

  

  Zhao Wang

  committed
  2319618

Commits on Apr 28, 2021

• code clean up, add type hint

  

  Zhao Wang

  committed
  5e12a29

• update dependency list

  

  Zhao Wang

  committed
  d2a00a0

Commits on Apr 30, 2021

• add support for multi GPU

  

  Zhao Wang

  committed
  659d003

• update post processing api

  

  Zhao Wang

  committed
  4f28496

• add close() fucntion back, was deleted by accident

  

  Zhao Wang

  committed
  99e9248

Commits on May 1, 2021

• remove process_frame, add type hint for __init__

  

  Zhao Wang

  committed
  1901c87

• remove cuda and trt context

  

  Zhao Wang

  committed
  d51139b

• make stride a local variable

  

  Zhao Wang

  committed
  07af9dd

• update type hint for batch_predict

  

  Zhao Wang

  committed
  f661509

• formatting

  

  Zhao Wang

  committed
  874414b

• formatting

  

  Zhao Wang

  committed
  7255b8c

Commits on May 3, 2021

• rename device_id to device_name

  

  Zhao Wang

  committed
  14d88f9

• _apply_box_norm will return int

  Show description for 1b93698
  

  BestDriverCN

  and

  apockill

  authored
  1b93698

• Merge branch 'tensorrt_backend' of github.com:opencv/open_vision_capsules into tensorrt_backend

  

  Zhao Wang

  committed
  5a372b1

• converting _apply_box_norm's return type

  

  Zhao Wang

  committed
  57d3eeb

• refactor parse_detection_results

  

  Zhao Wang

  committed
  ef70e0d

Commits on May 4, 2021

• fix import path, fix variable name, simplify code

  

  Zhao Wang

  committed
  a33112d

• simplify code

  

  Zhao Wang

  committed
  8fe0598

Commits on May 5, 2021

• change log level to INFO, cast score to float

  

  Zhao Wang

  committed
  1da5436