mvoelk / ssd_detectors

SSD-based object and text detection with Keras, SSD, DSOD, TextBoxes, SegLink, TextBoxes++, CRNN
MIT License
302 stars 85 forks source link

encode/decode error for tbpp #42

Closed cirgue closed 4 years ago

cirgue commented 4 years ago

I am trying to display the output from the tbpp model using real-world images. I have a set of results from the model, and am using the following code:

`from sl_utils import PriorUtil prior_util = PriorUtil(tbpp_model)

segment_threshold = 0.6; link_threshold = 0.25 res = prior_util.decode(results[0], segment_threshold, link_threshold) prior_util.plot_results(res)`

I get the following error:

ValueError: operands could not be broadcast together with shapes (76454,6) (76454,8)

The shape of 'results' is (1, 76454, 19)