Code

[Feature] NMS update#957

Code

Merged

ZwwWayne merged 31 commits intoopen-mmlab:masteropen-mmlab/mmcv:masterfrom SemyonBevzuk:nms_updateSemyonBevzuk/mmcv:nms_updateCopy head branch name to clipboardMay 31, 2021

+289-123

Commits

Commits on Apr 15, 2021

• Add score_threshold and max_num to NMS

  SemyonBevzuk

  committed
  8b1f267

• Fix codestyle

  SemyonBevzuk

  committed
  925fae1

Commits on Apr 16, 2021

• Merge branch 'master' into nms_update

  SemyonBevzuk

  committed
  3cee61c

• Fix codestyle

  SemyonBevzuk

  committed
  fd8b082

Commits on Apr 21, 2021

• Fix inds in nms

  SemyonBevzuk

  committed
  a6eae23

• Update nms docstring

  SemyonBevzuk

  committed
  ae469a7

• Move score_threshold and max_num arguments

  SemyonBevzuk

  committed
  23a465b

• Fix args order in docstring

  SemyonBevzuk

  committed
  61c7286

Commits on Apr 22, 2021

• Merge branch 'master' into nms_update

  RunningLeon

  committed
  de02886

• fix lint of c++ file

  RunningLeon

  committed
  8b4e7de

• Merge remote-tracking branch 'upstream/master' into nms_update

  SemyonBevzuk

  committed
  a9e45d6

Commits on Apr 23, 2021

• Remove torch.onnx.is_in_onnx_export() and add max_num to batched_nms for separate classes.

  SemyonBevzuk

  committed
  5c083b8

Commits on Apr 26, 2021

• Rewrote max_num handling in NMSop.symbolic

  SemyonBevzuk

  committed
  79e6941

Commits on Apr 27, 2021

• Added processing max_output_boxes_per_class when exporting to TensorRT

  SemyonBevzuk

  committed
  f8b1a7b

• Added score_threshold and max_num for NMS in test_onnx.py and test_tensorrt.py

  SemyonBevzuk

  committed
  9dba3c5

Commits on Apr 30, 2021

• Remove _is_value(max_num)

  SemyonBevzuk

  committed
  1355f5c

• Merge remote-tracking branch 'upstream/master' into nms_update

  SemyonBevzuk

  committed
  bc7fbb7

Commits on May 6, 2021

• fix ci errors with torch==1.3.1

  RunningLeon

  committed
  d2702b8

Commits on May 7, 2021

• Update test_batched_nms in test_nms.py

  SemyonBevzuk

  committed
  dc93be1

Commits on May 14, 2021

• Added tests for preprocess_onnx

  SemyonBevzuk

  committed
  95f1bbb

• Fix

  SemyonBevzuk

  committed
  6d415c9

• Moved 'test_tensorrt_preprocess.py' and 'preprocess', updated 'remove_tmp_file'.

  SemyonBevzuk

  committed
  9447eb6

Commits on May 17, 2021

• Update mmcv/tensorrt/__init__.py

  SemyonBevzuk

  committed
  e735646

Commits on May 18, 2021

• Fix segfault torch==1.3.1 (remove onnx.checker.check_model)

  SemyonBevzuk

  committed
  9d9c169

Commits on May 19, 2021

• Returned 'onnx.checker.check_model' with torch version check

  SemyonBevzuk

  committed
  8b03816

• Changed torch version from 1.3.1 to 1.4.0

  SemyonBevzuk

  committed
  5cd8a54

Commits on May 21, 2021

• update version check

  RunningLeon

  committed
  5f74a4c

• remove check for onnx

  RunningLeon

  committed
  7ba082e

Commits on May 24, 2021

• merge master and fix conflicts

  RunningLeon

  committed
  c9ac555

• Merge branch 'nms_update' of github.com:SemyonBevzuk/mmcv into nms_update

  RunningLeon

  committed
  f9fc45e

Commits on May 25, 2021

• merge master and resolve conflicts

  RunningLeon

  committed
  85576ea