What should I do if I want more labels?

NVIDIA-AI-IOT / CUDA-PointPillars

A project demonstrating how to use CUDA-PointPillars to deal with cloud points data from lidar.

Apache License 2.0

502 stars 148 forks source link

What should I do if I want more labels? #115

Open brillint opened 6 months ago

brillint commented 6 months ago

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 commented 5 months ago

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.