提交 f73d2c95 编写于 作者: W wjj19950828

support yolov8cls

上级 677cb91f
此差异已折叠。
......@@ -311,7 +311,9 @@ class PyTorchOpMapper():
last_name_segments = scope_name_part[index].split(".")
name_segments = scope_name_part[index + 1].split(".")
for j, name in enumerate(last_name_segments):
name_segments[j] = name
if j < len(name_segments) and name_segments[j] == name:
continue
name_segments.insert(j, name)
scope_name_part[index + 1] = ".".join(name_segments)
last_name = scope_name_part[-1]
name_segments = last_name.split(".")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册