leoneversberg/object-detection-pytorch
A short jupyter notebook to train a Faster R-CNN object detection model with PyTorch v2 using the COCO data format
This tool helps researchers and computer vision practitioners quickly train an object detection model to find specific items in images. You provide a collection of images where the objects of interest are already marked, and it produces a model capable of automatically identifying those objects in new, unseen images. It's designed for someone working with visual data who needs to automate the task of locating objects within pictures.
No commits in the last 6 months.
Use this if you have a dataset of images annotated in the COCO format and want a straightforward way to train a powerful object detection model using PyTorch.
Not ideal if you don't have pre-annotated images or need a solution that handles dataset creation and annotation from scratch.
Stars
7
Forks
—
Language
Jupyter Notebook
License
MIT
Category
Last pushed
Feb 21, 2024
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/leoneversberg/object-detection-pytorch"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
broadinstitute/keras-rcnn
Keras package for region-based convolutional neural networks (RCNNs)
alankbi/detecto
Build fully-functioning computer vision models with PyTorch
sovit-123/fasterrcnn-pytorch-training-pipeline
PyTorch Faster R-CNN Object Detection on Custom Dataset
kenshohara/video-classification-3d-cnn-pytorch
Video classification tools using 3D ResNet
lufficc/SSD
High quality, fast, modular reference implementation of SSD in PyTorch