AkshayLaddha943 / PersonDetection-RCNN-YOLO_COCODataset

Performance comparison of YOLO, FasterRCNN and SSD for Object Detection (Person)

PersonDetection-RCNN-YOLO_COCODataset

This repository contains an inferencing of pretrained object detectors (YOLOv5, SSD and Faster RCNN) for Person Detection on COCO Dataset

Pre-requisites

The project was performed with the following tools

• Python 3.11

• Pytorch 2.0

• VSCode

• COCO 2017

Dataset

Download the COCO dataset - (https://www.kaggle.com/datasets/awsaf49/coco-2017-dataset)

Results

Run the inference.py file using a pretrained model given from model.py

YOLOv5

Original Image