Skip to content

Commit a90915b

Browse files
committed
Parser field in nn archive default to YOLO
1 parent 06e8334 commit a90915b

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

tools/yolo/yolo26_exporter.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -122,14 +122,12 @@ def export_nn_archive(
122122
self.make_nn_archive(
123123
class_list=names,
124124
n_classes=self.model.model[-1].nc,
125-
parser="YOLOExtendedParser",
126125
encoding=encoding,
127126
)
128127
elif self.mode == SEGMENT_MODE:
129128
self.make_nn_archive(
130129
class_list=names,
131130
n_classes=self.model.model[-1].nc,
132-
parser="YOLOExtendedParser",
133131
n_prototypes=self.model.model[-1].nm,
134132
is_softmax=False, # E2E outputs are already sigmoided
135133
output_kwargs={
@@ -142,7 +140,6 @@ def export_nn_archive(
142140
self.make_nn_archive(
143141
class_list=names,
144142
n_classes=self.model.model[-1].nc,
145-
parser="YOLOExtendedParser",
146143
n_keypoints=self.model.model[-1].kpt_shape[0],
147144
output_kwargs={"keypoints_outputs": ["kpt_output"]},
148145
encoding=encoding,

0 commit comments

Comments
 (0)