Open martinazapletalova opened 2 months ago
At the end I've found solution in your medium article in comments that I have to add:
builder.spec.description.output.add()
before
builder.set_output(output_names=["raw_confidence", "raw_coordinates"], output_dims=[
(25200, numberOfClassLabels), (25200, 4)])
but now I have different problem when predicting:
Error compiling model: "compiler error: Error reading protobuf spec. validator error: Layer 'concat_confidence' of type 980 has 1 inputs but expects at least 2.
Hi, I'm trying to use your code in the colab notebook to add postprocessing to coreML yolo model. But I'm running into issue:
in step:
I've checked
builder.spec.description.output
and I can see only this:even though the layer with concat_coordinates is there b/c I'm getting
ValueError: Layer with name "concat_coordinates" has already been added. Please use a unique name.