Tech Stack
Python
PyTorch
Torchvision
Pascal VOC Dataset
Overview
This project implements a Faster R-CNN model for object detection. A custom dataset class was created for Pascal VOC 2007 annotations. The model was trained with data augmentations and evaluated using mAP, precision, and recall.
Features
- Custom dataset loader for Pascal VOC 2007 images + annotations.
- Training pipeline with checkpoint saving.
- Evaluation using mAP, precision, and recall metrics.