Visitar URL original
Addition of BUCTD Models by n-poulsen · Pull Request #2952 · DeepLabCut/DeepLabCut · GitHub
Skip to content

Addition of BUCTD Models #2952

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.

Sign up for GitHub

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

Merged
MMathisLab merged 114 commits into from
Apr 14, 2025
Merged

Addition of BUCTD Models #2952

Changes from 1 commit
Commits
Show all changes
114 commits
Select commit Hold shift + click to select a range
b57d849
add code of hrnet + coam backbone
Feb 2, 2024
0690899
refactor and finish hrnet+coam integration
Feb 2, 2024
c6e6668
add ctd configs
Feb 2, 2024
78ebb57
initial work on generative sampling
Feb 14, 2024
4503b87
buctd cont I
Mar 21, 2024
ec4aa87
prepare dataloader for loading cond pose with gen sampling
Mar 21, 2024
6c0cf37
adapt hrnet-coam output to hrnet code
Mar 21, 2024
704a0c6
put kpt encoders into modules
Mar 21, 2024
a53ba27
add kpt encoder to hrnet_coam
Mar 21, 2024
964dd5c
small additions
Mar 22, 2024
3e85fd1
init CTD inference
Mar 22, 2024
8c94c7b
test CTD training - I
Mar 22, 2024
107ea70
buctd training!
Mar 26, 2024
f8b7fa9
remove one for loop in _get_condition_matrix
Mar 27, 2024
9bb4ae6
optim trials for cond kpt encoding
Mar 28, 2024
ac29688
add ctd evaluation based on loaded bu predictions
Apr 7, 2024
bb31cc1
use only overlapping individuals for swapping error
Apr 8, 2024
ee7918f
add marmoset bu path
Apr 22, 2024
b83e8ec
Merge branch 'main' into lucas/buctd_integration
Apr 22, 2024
e769277
scale cond_kpts in collate
Apr 22, 2024
006b13a
switch back to original function for creating conditional matrix
Apr 24, 2024
1ccf705
update benchmark scripts
May 22, 2024
a4973c9
add hflip to internal benchmark script
May 22, 2024
304f9c2
pad with black pixels instead of context for CTD + add margin for BU-…
May 22, 2024
0b4be31
add script for testing ctd performance with coco api
May 22, 2024
c93c1bd
make CTD test inference compatible with previous code
May 23, 2024
9c9ef50
add bu name to output path names for CTD test inference
May 23, 2024
4de92f9
add bu name to output path names for CTD test inference - fix
May 23, 2024
fbbaf13
update BUCTD training with new TD developments
May 30, 2024
5a432b5
merged main
Jun 24, 2024
783cf79
Merge branch 'pytorch_dlc' into lucas/buctd_integration
Dec 9, 2024
8f3b82d
finish merge w main -> works as before
Jun 27, 2024
e930274
get_pose_inference_runner for CTD
Dec 9, 2024
0773924
lastest buctd devs
Jan 15, 2025
0609cc4
Merge branch 'pytorch_dlc' into lucas/buctd_v2
Jan 15, 2025
e432b9f
import fix
Jan 15, 2025
699a25f
func name fix
Jan 15, 2025
c792c9c
bug fix: pose postprocessor not built
Jan 15, 2025
1b723c2
bug fix: evaluate and inference
Jan 17, 2025
7cd1203
fix cropped_size variable and hacky test loading
Jan 17, 2025
8153c08
Merge branch 'lucas/buctd_v2' of github.com:AlexEMG/DLCdev into lucas…
Jan 17, 2025
c012941
Fixes to evaluation/inference on empty conditions
Feb 6, 2025
0be7e2e
added expected format
Feb 6, 2025
dd52784
Merge branch 'pytorch_dlc' into lucas/buctd_v2
Feb 6, 2025
46c213c
black for CoAM module
Feb 7, 2025
a521ff3
add conditional prenet draft
Mar 18, 2025
b0dcefd
add config files for hrnet+prenet
Mar 19, 2025
86724c4
add draft stacked keypoint enc
Mar 19, 2025
c58ddbe
updated code for CrowdPose/COCO OKS sigma
Mar 19, 2025
14b3eae
fix prenet details
Mar 20, 2025
a1aff5c
unfreeze bn stats of prenet+hrnet
Mar 20, 2025
77ee4cf
fix stacked keypoint encoding
Mar 24, 2025
c5d271a
Merge branch 'public' into lucas/buctd_v2
Mar 25, 2025
d4ed35c
improved loading conditions from files
Mar 26, 2025
dc7f1f0
removed debug prints
Mar 26, 2025
2c9a1b9
removed some debug comments
Mar 26, 2025
e095950
small clean up ctd configs
Mar 26, 2025
b6d99fe
improved loading conditions
Mar 26, 2025
4460b63
add config for rtmpose+prenet
Mar 27, 2025
a26d015
add img_size to hrnet_coam config
Mar 27, 2025
369048d
First implementation CTDInferenceRunner
Mar 27, 2025
583cfb2
Merge branch 'public' into lucas/buctd_v2
Mar 27, 2025
a9ea341
clean cond enc; extend ctd configs; add ctd related funcs to data loa…
Mar 27, 2025
6045f1a
Merge branch 'lucas/buctd_v2' of github.com:AlexEMG/DLCdev into lucas…
Mar 27, 2025
cf424ba
add config for humans and fix rtmpose ctd config
Mar 27, 2025
3fa99cb
fix config human
Mar 27, 2025
e568ca5
small fixes
Mar 27, 2025
7b100ce
improved CTD model config code
Mar 28, 2025
4ef5b47
implement re-scoring for CTD
Mar 28, 2025
d34aece
add TD aug for CTD models
Mar 28, 2025
e80bb5b
fix evaluation and debug print
Mar 28, 2025
19a3258
fixes to video inference
Mar 28, 2025
5ebcbdc
add gen sampling params for crowdpose
Mar 28, 2025
4815020
Merge branch 'lucas/buctd_v2' of github.com:AlexEMG/DLCdev into lucas…
Mar 28, 2025
86b6704
small fix config human ctd
Mar 28, 2025
6f55b9f
bug fix: dataset idv IDs
Mar 28, 2025
38b1a26
Merge branch 'lucas/buctd_v2' of github.com:AlexEMG/DLCdev into lucas…
Mar 28, 2025
5a3f227
bug fix: dataset idv IDs
Mar 28, 2025
2b1205f
Merge branch 'lucas/buctd_v2' of github.com:AlexEMG/DLCdev into lucas…
Mar 28, 2025
edbc25a
fix bug when no keypoints are visible
Mar 28, 2025
eee62d7
Merge branch 'lucas/buctd_v2' of github.com:AlexEMG/DLCdev into lucas…
Mar 28, 2025
ab75ce9
fix dataloader: trim area and bboxes
Mar 28, 2025
0120285
Merge branch 'lucas/buctd_v2' of github.com:AlexEMG/DLCdev into lucas…
Mar 28, 2025
29d9663
implemented CTD inference
Mar 28, 2025
9e157bf
add coam config for crowdpose
Mar 28, 2025
24c6db5
Merge branch 'lucas/buctd_v2' of github.com:AlexEMG/DLCdev into lucas…
Mar 28, 2025
43bda99
First implementation of CTD tracking
Apr 1, 2025
24b06ba
Add `ctd_conditions` argument to `deeplabcut.create_training_dataset(…
maximpavliv Apr 1, 2025
2453441
implement user-friendly CTD inference (#268)
n-poulsen Apr 9, 2025
cb65d69
bug fixes and improvements to default configs
Apr 10, 2025
63c6ce2
update LR for CTD models
Apr 10, 2025
8cbf38a
rename Task.CTD to Task.COND_TOP_DOWN (#270)
n-poulsen Apr 10, 2025
e5d2d28
remove unused code
Apr 10, 2025
12cf3fa
remove unused code
Apr 10, 2025
8140467
add link to COLAB in maDLC_UserGuide
Apr 11, 2025
2a43052
Merge branch 'main' into lucas/buctd_v2
Apr 11, 2025
f54cc47
update BUCTD colab
Apr 11, 2025
4ff3d95
fix COLAB rendering
Apr 11, 2025
bd2d14d
fix COLAB rendering
Apr 11, 2025
28ff47f
fix COLAB rendering
Apr 11, 2025
a04a9bd
fix pip install in COLAB
Apr 11, 2025
c42fe22
skip notebooks for codespell
Apr 11, 2025
1d92522
install deeplabcut with --pre in colab
Apr 11, 2025
b0dae22
fix tests
Apr 11, 2025
9696a4b
Merge branch 'main' into lucas/buctd_v2
MMathisLab Apr 11, 2025
9eec042
fix failing tests on windows
Apr 11, 2025
a42c61d
fix tests windows
Apr 11, 2025
1d67089
Update architectures.md
AlexEMG Apr 14, 2025
d6b3f74
deal with absolute windows paths
Apr 14, 2025
0fc92a0
Version update
AlexEMG Apr 14, 2025
6a1e0a9
fix when paths are converted to windows
Apr 14, 2025
3900605
Merge branch 'main' into lucas/buctd_v2
Apr 14, 2025
fb7c647
fix windows tests
Apr 14, 2025
44a54fb
Merge branch 'main' into lucas/buctd_v2
MMathisLab Apr 14, 2025
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
Merge branch 'main' into lucas/buctd_v2
  • Loading branch information
@MMathisLab
MMathisLab authored Apr 14, 2025
commit 44a54fb2360fcd310dc6a2cc02608c7d3c7cbaa9

This merge commit was added into this branch cleanly.

There are no new changes to show, but you can still view the diff.