Add https://www.reddit.com/r/Ultralytics/ badge (#13284)
* Refactor code for speed and clarity * Update README.zh-CN.md Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> --------- Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
1 parent
b40781b3e3
commit
c5ffbbf1c3
20 files changed
+164
-73
No files matched your search
@@ -12,7 +12,8 @@ from .general import crop_mask
|
||||
|
||||
|
||||
class ComputeLoss:
|
||||
# Compute losses
|
||||
"""Computes the YOLOv5 model's loss components including classification, objectness, box, and mask losses."""
|
||||
|
||||
def __init__(self, model, autobalance=False, overlap=False):
|
||||
"""Initializes the compute loss function for YOLOv5 models with options for autobalancing and overlap
|
||||
handling.
|
||||
|
||||
Reference in new issue
Block a user