2D Object Detection | Model Comparisons

1) Dataset

• I took dataset from the NuScene, created annotation on it from the Roboflow, a web based annotations software and tried 3 different algorithms on generated dataset. Here is the information of my dataset in visual form.

• Information graphs for labels and labels-correlations.

2) Scaled YOLOv4

• Precision Recall Curve and result graphs for Scaled YOLOv4

3) YOLOv5

• Precision Recall Curve and result graphs for YOLOv5

4) Detectron2

• For this I do not have the data output but I have the COCO evaluation matrix, which is below.

5) Results

• Scaled YOLOv4

• YOLOv5

• Detectron2