alexppppp/keypoint_rcnn_training_pytorch
How to Train a Custom Keypoint Detection Model with PyTorch (Article on Medium)
This project helps computer vision practitioners train a custom model to identify specific points on objects within images. You input a dataset of images annotated with keypoint locations, and it outputs a trained model capable of detecting these keypoints on new, unseen images. This is ideal for researchers, engineers, or anyone working with visual data who needs to precisely locate features like human joints, facial landmarks, or industrial component points.
No commits in the last 6 months.
Use this if you need to build a specialized computer vision system that can accurately pinpoint specific, defined locations on objects in your images.
Not ideal if you are looking for an out-of-the-box solution for general object detection or image classification without the need for precise keypoint localization.
Stars
92
Forks
34
Language
Python
License
MIT
Category
Last pushed
Apr 07, 2023
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/alexppppp/keypoint_rcnn_training_pytorch"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
KaiyangZhou/deep-person-reid
Torchreid: Deep learning person re-identification in PyTorch.
Cysu/open-reid
Open source person re-identification library in python
mangye16/ReID-Survey
Deep Learning for Person Re-identification: A Survey and Outlook
jian667/face-dataset
Face related datasets
tlpss/keypoint-detection
2D keypoint detection with Pytorch Lightning and wandb