Fix export of trained seg models (#121)

luxonis · Nov 27, 2024 · 875ee41 · 875ee41
1 parent 046a345
commit 875ee41
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "tools"
-version = "0.1.3"
+version = "0.1.4"
 description = "Converter for YOLO models into .ONNX format."
 readme = "README.md"
 requires-python = ">=3.8"

diff --git a/tools/modules/heads.py b/tools/modules/heads.py
@@ -482,7 +482,6 @@ def __init__(self, old_segment, use_rvc2):
         self.nm = old_segment.nm  # number of masks
         self.npr = old_segment.npr  # number of protos
         self.proto = old_segment.proto  # protos
-        self.detect = old_segment.detect
         self.cv4 = old_segment.cv4
 
     def forward(self, x):