Code

Update of PoseEstimation model#872

Code

Merged

BloodAxe merged 82 commits intomasterDeci-AI/super-gradients:masterfrom feature/SG-707-rescoring-netDeci-AI/super-gradients:feature/SG-707-rescoring-netCopy head branch name to clipboardMay 4, 2023

+1,069-281

Commits

Commits on Apr 12, 2023

• Rescoring net

  BloodAxe

  committed
  a9ad6ae

Commits on Apr 13, 2023

• Added docs

  BloodAxe

  committed
  11c3e3f

• Added docs

  BloodAxe

  committed
  c3374be

Commits on Apr 17, 2023

• Tune recipe

  BloodAxe

  committed
  78f0000

• Reduce bs for validation loader

  BloodAxe

  committed
  2e94101

• Reduce bs for validation loader

  BloodAxe

  committed
  cee3b0c

• Do not store predictions on GPU

  BloodAxe

  committed
  95da0fa

• Do not store predictions on GPU

  BloodAxe

  committed
  cd33fe0

• Do not store predictions on GPU

  BloodAxe

  committed
  ada17b6

• Added ImageKeypointMatchingResult dataclass as return value for compute_img_keypoint_matching

  BloodAxe

  committed
  8507ca7

• Script to generate trianing data (WIP)

  BloodAxe

  committed
  e24c74b

• Do not store predictions on GPU

  BloodAxe

  committed
  54c2d5b

• Lower LR

  BloodAxe

  committed
  642ba9c

• Merge remote-tracking branch 'origin/feature/SG-707-rescoring-net' into feature/SG-707-rescoring-net

  BloodAxe

  committed
  b7b11d2

• Merge branch 'master' into feature/SG-707-rescoring-net

  BloodAxe

  committed
  86034c8

• Added check to ensure that person area is positive

  BloodAxe

  committed
  602b8a0

• Added check to ensure that person area is positive

  BloodAxe

  committed
  37cedd0

• Save all samples to enable us computing mAP score during rescore model training

  BloodAxe

  committed
  a605817

• Save all samples to enable us computing mAP score during rescore model training

  BloodAxe

  committed
  e55b94b

• Fix edge case when no predictions are made

  BloodAxe

  committed
  16a1daf

Commits on Apr 18, 2023

• Rescoring training

  BloodAxe

  committed
  64033a5

Commits on Apr 19, 2023

• Fix convert_to_tensor

  BloodAxe

  committed
  1b98bce

Commits on Apr 21, 2023

• High LR config

  BloodAxe

  committed
  1b081ec

• Disable ema

  BloodAxe

  committed
  f5670f2

Commits on Apr 25, 2023

• Merge branch 'master' into feature/SG-707-rescoring-net

  Show description for 58de569
  BloodAxe

  committed
  58de569

• Fix rescoring training

  BloodAxe

  committed
  0c6010b

• Recipe for DDRNet39

  BloodAxe

  committed
  8cdc0f0

• Recipe for DDRNet39

  BloodAxe

  committed
  3df3372

• Recipe for DEKR with QFL

  BloodAxe

  committed
  5702b7b

• Preprocessing script to save data on train dataset

  BloodAxe

  committed
  414af0a

Commits on Apr 26, 2023

• Update rescoring dataset to load data from Picke

  BloodAxe

  committed
  bdc553e

Commits on Apr 27, 2023

• Update recipe

  BloodAxe

  committed
  aac0ab8

• Increase num people to 30

  Show description for 3dc2460
  BloodAxe

  committed
  3dc2460

• New dataset params

  BloodAxe

  committed
  bce0b8f

• New dataset params

  BloodAxe

  committed
  54ff914

• New dataset params

  BloodAxe

  committed
  da3ea12

• New dataset params

  BloodAxe

  committed
  dea1231

• New dataset params

  BloodAxe

  committed
  a726ae6

• New dataset params

  BloodAxe

  committed
  a410aeb

• New dataset params

  BloodAxe

  committed
  f94f349

• Update instructions and recipes to generate rescoring data

  BloodAxe

  committed
  82db6d1

• Update recipies to support TTA

  BloodAxe

  committed
  a1b40c3

Commits on Apr 28, 2023

• Use BCE instead of MSE while training rescoring network. It gives higher mAP score

  BloodAxe

  committed
  cd0ae5c

Commits on Apr 29, 2023

• coco2017_pose_dekr_w32_no_dc_high_lr_qfl_hard_augs_no_ema

  BloodAxe

  committed
  c8e369c

• coco2017_pose_dekr_w32_no_dc_high_lr_qfl_hard_augs_no_ema

  BloodAxe

  committed
  5f7de13

• coco2017_pose_dekr_w32_no_dc_high_lr_qfl_hard_augs_no_ema

  BloodAxe

  committed
  332f90a

• fp16

  BloodAxe

  committed
  870603b

• fp16

  BloodAxe

  committed
  179141f

• fp16

  BloodAxe

  committed
  184e586

Commits on May 1, 2023

• Merge branch 'master' into feature/SG-707-rescoring-net

  BloodAxe

  committed
  7deb9a1

• Update recipies to train pose estimation

  BloodAxe

  committed
  184e332

• Update AP scores for pose estimation model

  BloodAxe

  committed
  7162be0

• Merge branch 'master' into feature/SG-707-rescoring-net

  BloodAxe

  authored
  07209a6

• Adding tests for TTA & Rescoring

  BloodAxe

  committed
  897b63a

• torch.float -> torch.float32

  BloodAxe

  committed
  2c59103

• Adding docs

  BloodAxe

  committed
  bd67954

• Adding docs

  BloodAxe

  committed
  345d3e4

• Remove test_rescoring_net.py

  BloodAxe

  committed
  4a3ceb8

• Fix typo in script -> scripts

  BloodAxe

  committed
  4dd1bbf

• Changed implementation of pose_nms to return empty arrays of shape 0,17,3

  BloodAxe

  committed
  c030d3f

• Rename integration test file for pose estimation

  BloodAxe

  committed
  63a9dcc

• Rename integration test file for pose estimation

  BloodAxe

  committed
  f9aea7b

• Rename integration test file for pose estimation

  BloodAxe

  committed
  84c7799

• Rename integration test file for pose estimation

  BloodAxe

  committed
  6ea8486

• Rename integration test file for pose estimation

  BloodAxe

  committed
  49ed049

Commits on May 2, 2023

• Update script config to generate rescoring training data

  BloodAxe

  committed
  dcfc6ee

• Update script config to generate rescoring training data

  BloodAxe

  committed
  9960f1e

• Merge branch 'master' into feature/SG-707-rescoring-net

  BloodAxe

  authored
  b3c778d

• Update configs names

  BloodAxe

  committed
  d8f79da

• Bugfix in post-processing callback for rescoring

  BloodAxe

  committed
  e9eb3ba

• Merge remote-tracking branch 'origin/feature/SG-707-rescoring-net' into feature/SG-707-rescoring-net

  BloodAxe

  committed
  351476a

• Merge master

  BloodAxe

  committed
  bea7154

• Merge branch 'master' into feature/SG-707-rescoring-net

  BloodAxe

  authored
  1affd39

• Rename PoseEstimationDatasetIntegrationTest to PoseEstimationModelsIntegrationTest

  BloodAxe

  committed
  ed1be3e

• Merge remote-tracking branch 'origin/feature/SG-707-rescoring-net' into feature/SG-707-rescoring-net

  BloodAxe

  committed
  122f113

• Update models url's

  BloodAxe

  committed
  9ae5060

• Merge branch 'master' into feature/SG-707-rescoring-net

  BloodAxe

  authored
  225811c

Commits on May 3, 2023

• Merge branch 'master' into feature/SG-707-rescoring-net

  BloodAxe

  authored
  8f1f42b

Commits on May 4, 2023

• Merge branch 'master' into feature/SG-707-rescoring-net

  BloodAxe

  authored
  405d66b

• Merge branch 'master' into feature/SG-707-rescoring-net

  BloodAxe

  authored
  47da0ee

• Added direct link to test

  BloodAxe

  committed
  3280bfb

• Merge branch 'master' into feature/SG-707-rescoring-net

  BloodAxe

  authored
  e8a7905