Chiebot-Mirror
/
ultralytics
mirror of https://github.com/ultralytics/ultralytics.git
9
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Auto-format by https://ultralytics.com/actions

Browse Source
new-v11-head
UltralyticsAssistant 5 months ago
parent 160eca3a3a
commit 43cf5f977a
1 changed files with 7 additions and 8 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 3
      ultralytics/nn/modules/head.py

3
ultralytics/nn/modules/head.py
Unescape View File

@ -592,7 +592,7 @@ class v10Detect(Detect):
class v11Detect(Detect): class v11Detect(Detect):
""" """
v11 Detection head V11 Detection head.
Args: Args:
nc (int): Number of classes. nc (int): Number of classes.
@ -602,7 +602,6 @@ class v11Detect(Detect):
__init__(self, nc=80, ch=()): Initializes the v11Detect object. __init__(self, nc=80, ch=()): Initializes the v11Detect object.
forward(self, x): Performs forward pass of the v11Detect module. forward(self, x): Performs forward pass of the v11Detect module.
bias_init(self): Initializes biases of the Detect module. bias_init(self): Initializes biases of the Detect module.
""" """
def __init__(self, nc=80, ch=()): def __init__(self, nc=80, ch=()):

Write Preview
Loading…
Cancel
Save