A PyTorch landmarks-only library with 100+ data augmentations, training and inference
English | δΈζζζ‘£
π€ Introduction
torchlm is a PyTorch landmarks-only library with 100+ data augmentations, support training and inference. torchlm is aims at only focus on any landmark detection, such as face landmarks, hand keypoints and body keypoints, etc. It provides 30+ native data augmentations and can bind with 80+ transforms from torchvision and albumentations, no matter the input is a np.ndarray or a torch Tensor, torchlm will automatically be compatible with different data types and then wrap it back to the original type through a autodtype wrapper. Further, torchlm will add modules for training and inference in the future.
β€οΈ Star πππ» this repo to support me if it does any helps to you, thanks