Skip to content

Detection of traffic signs and potholes by Yolo8 and Fast-RCNN

Notifications You must be signed in to change notification settings

alinzh/road_detection

Repository files navigation

Road-detection: detection of traffic signs and potholes by ML-models

Implementation of detection by transfer learning with popular Yolo8 and Fast-RCNN from Detectrons zoo. Training takes place on a custom dataset assembled from several.

screenshot

The dataset includes 157 classes:

  • 156 - different types of Russian traffic signs
  • 1 - potholes

The repository contains a script for assembling datasets and bringing them to required annotation format.

Both datasets are available at the following links:

Download weights for trained Yolo8m from here.

Summary

Not bad! summary

About

Detection of traffic signs and potholes by Yolo8 and Fast-RCNN

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages