Skip to content

aybora/yolov5Loss

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 

Repository files navigation

yolov5Loss-OGAM

This work modifies original YOLOv5 work for hard example mining via our proposed combination of Balanced Focal Loss and Loss Rank Mining (LRM) approach. Original loss.py file is modified for LRM implementation.

Tested on YOLOv5s of v5.0 release and verified on YOLOv5l of v6.0 release.

Use LRM_ignore flag for LRM activation and fl_gamma and obj flags for Balanced Focal Loss.

Related paper: https://arxiv.org/abs/2202.13080

Please consider citing our paper and YOLOv5 while using our algorithm.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages