Skip to content

Training YOLO algorithm on a custom dataset which has multi-class for Rail Fault Detection on Google Colab. Then used this detector as a teacher and labeled all unseen image to create pseudo labeled images. Finally, pseudo labeled images used to train the detector again to boost model performance.

License

Notifications You must be signed in to change notification settings

ridvanozdemir/Rail_Fault_Detection_with_YOLOv4

About

Training YOLO algorithm on a custom dataset which has multi-class for Rail Fault Detection on Google Colab. Then used this detector as a teacher and labeled all unseen image to create pseudo labeled images. Finally, pseudo labeled images used to train the detector again to boost model performance.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published