File tree Expand file tree Collapse file tree 1 file changed +0
-3
lines changed
Expand file tree Collapse file tree 1 file changed +0
-3
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments