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

Update of PoseEstimation model #872

Merged
merged 82 commits into from
May 4, 2023
Merged
Changes from 1 commit
Commits
Show all changes
82 commits
Select commit Hold shift + click to select a range
a9ad6ae
Rescoring net
BloodAxe Apr 12, 2023
11c3e3f
Added docs
BloodAxe Apr 13, 2023
c3374be
Added docs
BloodAxe Apr 13, 2023
78f0000
Tune recipe
BloodAxe Apr 17, 2023
2e94101
Reduce bs for validation loader
BloodAxe Apr 17, 2023
cee3b0c
Reduce bs for validation loader
BloodAxe Apr 17, 2023
95da0fa
Do not store predictions on GPU
BloodAxe Apr 17, 2023
cd33fe0
Do not store predictions on GPU
BloodAxe Apr 17, 2023
ada17b6
Do not store predictions on GPU
BloodAxe Apr 17, 2023
8507ca7
Added ImageKeypointMatchingResult dataclass as return value for compu…
BloodAxe Apr 17, 2023
e24c74b
Script to generate trianing data (WIP)
BloodAxe Apr 17, 2023
54c2d5b
Do not store predictions on GPU
BloodAxe Apr 17, 2023
642ba9c
Lower LR
BloodAxe Apr 17, 2023
b7b11d2
Merge remote-tracking branch 'origin/feature/SG-707-rescoring-net' in…
BloodAxe Apr 17, 2023
86034c8
Merge branch 'master' into feature/SG-707-rescoring-net
BloodAxe Apr 17, 2023
602b8a0
Added check to ensure that person area is positive
BloodAxe Apr 17, 2023
37cedd0
Added check to ensure that person area is positive
BloodAxe Apr 17, 2023
a605817
Save all samples to enable us computing mAP score during rescore mode…
BloodAxe Apr 17, 2023
e55b94b
Save all samples to enable us computing mAP score during rescore mode…
BloodAxe Apr 17, 2023
16a1daf
Fix edge case when no predictions are made
BloodAxe Apr 17, 2023
64033a5
Rescoring training
BloodAxe Apr 18, 2023
1b98bce
Fix convert_to_tensor
BloodAxe Apr 19, 2023
1b081ec
High LR config
BloodAxe Apr 21, 2023
f5670f2
Disable ema
BloodAxe Apr 21, 2023
58de569
Merge branch 'master' into feature/SG-707-rescoring-net
BloodAxe Apr 25, 2023
0c6010b
Fix rescoring training
BloodAxe Apr 25, 2023
8cdc0f0
Recipe for DDRNet39
BloodAxe Apr 25, 2023
3df3372
Recipe for DDRNet39
BloodAxe Apr 25, 2023
5702b7b
Recipe for DEKR with QFL
BloodAxe Apr 25, 2023
414af0a
Preprocessing script to save data on train dataset
BloodAxe Apr 25, 2023
bdc553e
Update rescoring dataset to load data from Picke
BloodAxe Apr 26, 2023
aac0ab8
Update recipe
BloodAxe Apr 27, 2023
3dc2460
Increase num people to 30
BloodAxe Apr 27, 2023
bce0b8f
New dataset params
BloodAxe Apr 27, 2023
54ff914
New dataset params
BloodAxe Apr 27, 2023
da3ea12
New dataset params
BloodAxe Apr 27, 2023
dea1231
New dataset params
BloodAxe Apr 27, 2023
a726ae6
New dataset params
BloodAxe Apr 27, 2023
a410aeb
New dataset params
BloodAxe Apr 27, 2023
f94f349
New dataset params
BloodAxe Apr 27, 2023
82db6d1
Update instructions and recipes to generate rescoring data
BloodAxe Apr 27, 2023
a1b40c3
Update recipies to support TTA
BloodAxe Apr 27, 2023
cd0ae5c
Use BCE instead of MSE while training rescoring network. It gives hig…
BloodAxe Apr 28, 2023
c8e369c
coco2017_pose_dekr_w32_no_dc_high_lr_qfl_hard_augs_no_ema
BloodAxe Apr 29, 2023
5f7de13
coco2017_pose_dekr_w32_no_dc_high_lr_qfl_hard_augs_no_ema
BloodAxe Apr 29, 2023
332f90a
coco2017_pose_dekr_w32_no_dc_high_lr_qfl_hard_augs_no_ema
BloodAxe Apr 29, 2023
870603b
fp16
BloodAxe Apr 29, 2023
179141f
fp16
BloodAxe Apr 29, 2023
184e586
fp16
BloodAxe Apr 29, 2023
7deb9a1
Merge branch 'master' into feature/SG-707-rescoring-net
BloodAxe May 1, 2023
184e332
Update recipies to train pose estimation
BloodAxe May 1, 2023
7162be0
Update AP scores for pose estimation model
BloodAxe May 1, 2023
07209a6
Merge branch 'master' into feature/SG-707-rescoring-net
BloodAxe May 1, 2023
897b63a
Adding tests for TTA & Rescoring
BloodAxe May 1, 2023
2c59103
torch.float -> torch.float32
BloodAxe May 1, 2023
bd67954
Adding docs
BloodAxe May 1, 2023
345d3e4
Adding docs
BloodAxe May 1, 2023
4a3ceb8
Remove test_rescoring_net.py
BloodAxe May 1, 2023
4dd1bbf
Fix typo in script -> scripts
BloodAxe May 1, 2023
c030d3f
Changed implementation of pose_nms to return empty arrays of shape 0,…
BloodAxe May 1, 2023
63a9dcc
Rename integration test file for pose estimation
BloodAxe May 1, 2023
f9aea7b
Rename integration test file for pose estimation
BloodAxe May 1, 2023
84c7799
Rename integration test file for pose estimation
BloodAxe May 1, 2023
6ea8486
Rename integration test file for pose estimation
BloodAxe May 1, 2023
49ed049
Rename integration test file for pose estimation
BloodAxe May 1, 2023
dcfc6ee
Update script config to generate rescoring training data
BloodAxe May 2, 2023
9960f1e
Update script config to generate rescoring training data
BloodAxe May 2, 2023
b3c778d
Merge branch 'master' into feature/SG-707-rescoring-net
BloodAxe May 2, 2023
d8f79da
Update configs names
BloodAxe May 2, 2023
e9eb3ba
Bugfix in post-processing callback for rescoring
BloodAxe May 2, 2023
351476a
Merge remote-tracking branch 'origin/feature/SG-707-rescoring-net' in…
BloodAxe May 2, 2023
bea7154
Merge master
BloodAxe May 2, 2023
1affd39
Merge branch 'master' into feature/SG-707-rescoring-net
BloodAxe May 2, 2023
ed1be3e
Rename PoseEstimationDatasetIntegrationTest to PoseEstimationModelsIn…
BloodAxe May 2, 2023
122f113
Merge remote-tracking branch 'origin/feature/SG-707-rescoring-net' in…
BloodAxe May 2, 2023
9ae5060
Update models url's
BloodAxe May 2, 2023
225811c
Merge branch 'master' into feature/SG-707-rescoring-net
BloodAxe May 2, 2023
8f1f42b
Merge branch 'master' into feature/SG-707-rescoring-net
BloodAxe May 3, 2023
405d66b
Merge branch 'master' into feature/SG-707-rescoring-net
BloodAxe May 4, 2023
47da0ee
Merge branch 'master' into feature/SG-707-rescoring-net
BloodAxe May 4, 2023
3280bfb
Added direct link to test
BloodAxe May 4, 2023
e8a7905
Merge branch 'master' into feature/SG-707-rescoring-net
BloodAxe May 4, 2023
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
New dataset params
BloodAxe committed Apr 27, 2023

Verified

This commit was signed with the committer’s verified signature.
BloodAxe Eugene Khvedchenya
commit a410aeb29e2af3c58885f7b0e514f7e220d397cd
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
defaults:
- coco_pose_estimation_dataset_params
- _self_
-

train_dataset_params:
min_instance_area: 64