From 0f4d9adbf20b1d56d575fe3460e440989650a89f Mon Sep 17 00:00:00 2001 From: HonzaCuhel Date: Tue, 26 Nov 2024 13:52:52 +0100 Subject: [PATCH] Fix export of trained seg models --- pyproject.toml | 2 +- tools/modules/heads.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index ba715b4..8bcd8e0 100644 --- 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 index 81cb81c..664af3c 100644 --- 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):