mvoelk/ssd_detectors

encode/decode error for tbpp

cirgue opened this issue · 0 comments

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)