$((CLASSES+5))

PINTO0309 · Oct 4, 2023 · 38c1ed5 · 38c1ed5
1 parent 6d06b97
commit 38c1ed5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/307_YOLOv7/post_process_gen_tools/convert_script.sh b/307_YOLOv7/post_process_gen_tools/convert_script.sh
@@ -158,7 +158,7 @@ sio4onnx \
 --input_onnx_file_path 12_nms_${MODEL_NAME}_${BOXES}_batch.onnx \
 --output_onnx_file_path 12_nms_${MODEL_NAME}_${BOXES}_batch.onnx \
 --input_names "predictions" \
---input_shapes "batch" ${BOXES} 85 \
+--input_shapes "batch" ${BOXES} $((CLASSES+5)) \
 --output_names "x1y1x2y2" \
 --output_names "workaround_mul_out" \
 --output_shapes "batch" ${BOXES} 4 \