What should I do if I want more labels?

[brillint]

trafficstars

when I have 8 labels,it returns: onnx.onnx_cpp2py_export.shape_inference.InferenceError: [ShapeInferenceError] (op_type:Transpose, node name: Transpose_309): [ShapeInferenceError] Inferred shape and existing shape differ in dimension 3: (128) vs (32)

[ybcc2015]

You need to modify the shape of cls_preds , box_preds , dir_cls_preds according to the your grid size and number of categories, in the simplify_postprocess() func.