Skip to content

Commit d738a2c

Browse files
committed
fix default top down dynamic cropping parameters
1 parent c6a1f69 commit d738a2c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dlclive/pose_estimation_pytorch/dynamic_cropping.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -261,13 +261,13 @@ class TopDownDynamicCropper(DynamicCropper):
261261
def __init__(
262262
self,
263263
top_down_crop_size: tuple[int, int] = (256, 256),
264-
patch_counts: tuple[int, int] = (4, 3),
264+
patch_counts: tuple[int, int] = (3, 2),
265265
patch_overlap: int = 50,
266-
min_bbox_size: tuple[int, int] = (100, 100),
267-
threshold: float = 0.6,
266+
min_bbox_size: tuple[int, int] = (50, 50),
267+
threshold: float = 0.25,
268268
margin: int = 10,
269269
min_hq_keypoints: int = 2,
270-
bbox_from_hq: bool = False,
270+
bbox_from_hq: bool = True,
271271
store_crops: bool = False,
272272
**kwargs,
273273
) -> None:

0 commit comments

Comments
 (0)