search

tlpss / keypoint-detection

2D keypoint detection with Pytorch Lightning and wandb
MIT License
70 stars 9 forks source link

added argument for max amount of keypoints #18

Closed lucasvandijck closed 1 year ago

lucasvandijck commented 1 year ago

Added simple extra argument to choose how many keypoints to predict. By default this arg is set to -1 meaning all peaks are turned into predictions.

lucasvandijck commented 1 year ago

-1 is not an ideal default, maybe set it back to 20