/mask_rcnn_vs_yolov8

A comaprison of mask_rcnn vs yolov8 for segmentation task in applied machine learning

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Comparison of Mask RCNN vs Yolov8

  • The goal of this assignment is train both models on custom annotated dataset.

  • Take photos of your environment of two or more objects. (at least 100 instances between all objects)

  • Annotate them on roboflow.

  • Train a Mask RCNN model using detectron2

  • Train Yolov8 the smallest size

  • Evaluate both models based on mAP and speed and size.

Taking pictures

A notebook and airpods were used for segmentation task

instagram instagram

Roboflow annotations

Screenshots from roboflow

Airpods

image

Book

Screenshot (78)

Sample outputs

Mask RCNN detectron2

image image

Yolov8

preds

Comparison

Mean Average Precision

  • Mask RCNN: 79.77%
  • Yolov8: 97.4%

Speed

  • Yolov8's significant speed advantage provides numerous benefits, even considering its size.
  • Training Yolov8 for 20 epochs takes approximately 3-4 minutes, whereas MASK RCNN requires approximately 55-56 minutes.

Size

  • Mask RCNN: 335.82 Mb
  • Yolov8: 22.79 Mb

Colab Notebook

Open In Colab