Skip to content

Add Functions for Visualizing COCO Predictions and Ground Truth #2792

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
n-poulsen merged 20 commits into from
Dec 2, 2024
Merged

Add Functions for Visualizing COCO Predictions and Ground Truth #2792

Changes from 1 commit
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
598d02b
Fix: correct the early return error when save_epochs=1 and delelte th…
xiu-cs Nov 21, 2024
76e10b4
Update deeplabcut/pose_estimation_pytorch/runners/snapshots.py
xiu-cs Nov 22, 2024
0b64247
Merge branch 'pytorch_dlc' into ti/snapshots
xiu-cs Nov 22, 2024
f670407
black -> snapshots.py
xiu-cs Nov 22, 2024
da99c66
add plot_gt_and_prediction function
xiu-cs Nov 21, 2024
2643d6b
delete the initial attempt at function
xiu-cs Nov 21, 2024
3aa926b
add def visualize_coco_predictions(*)
xiu-cs Nov 21, 2024
f0a24f7
isort and black -> evaluate.py
xiu-cs Nov 25, 2024
1fdd31e
modify the output_dir
xiu-cs Nov 26, 2024
bec1cff
Revert "black -> snapshots.py"
xiu-cs Nov 26, 2024
17fe504
revert changes to snapshots
xiu-cs Nov 26, 2024
bf80861
optimize the input params
xiu-cs Nov 26, 2024
694a814
rename the function and fix the multi-animal ploting problem
xiu-cs Nov 26, 2024
a731901
use num_individuals
xiu-cs Nov 26, 2024
74e8954
optimize the codes
xiu-cs Nov 26, 2024
c4b5a4a
add show_ground_truth param for future use
xiu-cs Nov 26, 2024
e33a745
black
xiu-cs Nov 26, 2024
31e6b32
Update deeplabcut/pose_estimation_pytorch/apis/evaluate.py
xiu-cs Dec 2, 2024
88949d1
Update deeplabcut/pose_estimation_pytorch/apis/evaluate.py
xiu-cs Dec 2, 2024
ead180e
Update deeplabcut/pose_estimation_pytorch/apis/evaluate.py
xiu-cs Dec 2, 2024
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
Update deeplabcut/pose_estimation_pytorch/apis/evaluate.py 
Co-authored-by: n-poulsen <[email protected]>
  • Loading branch information
@xiu-cs @n-poulsen
xiu-cs and n-poulsen authored Dec 2, 2024
commit ead180e1fdcf29fd74138bbc97f2182d44b82b5a
2 changes: 0 additions & 2 deletions deeplabcut/pose_estimation_pytorch/apis/evaluate.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -202,8 +202,6 @@ def visualize_predictions(
output_dir: Path to save visualization outputs.
Defaults to "predictions_visualizations"

draw_skeleton: Whether to draw skeleton connections between keypoints

num_samples: Number of images to visualize. If None, processes all images

random_select: If True, randomly samples images; if False, uses first N images
Expand Down