Hello !
I am getting the following error while implementing the yolo_filter_box
:
TypeError Traceback (most recent call last)
in
3 boxes = tf.random.normal([19, 19, 5, 4], mean=1, stddev=4, seed = 1)
4 box_class_probs = tf.random.normal([19, 19, 5, 80], mean=1, stddev=4, seed = 1)
----> 5 scores, boxes, classes = yolo_filter_boxes(boxes, box_confidence, box_class_probs, threshold = 0.5)
6 print("scores[2] = " + str(scores[2].numpy()))
7 print("boxes[2] = " + str(boxes[2].numpy()))
in yolo_filter_boxes(boxes, box_confidence, box_class_probs, threshold)
56 ## (â 1 line)
57 print(type(box_class_scores))
â> 58 filtering_mask = box_class_scores >= threshold
59
60
/opt/conda/lib/python3.7/site-packages/tensorflow/python/ops/gen_math_ops.py in greater_equal(x, y, name)
4053 _result = pywrap_tfe.TFE_Py_FastPathExecute(
4054 _ctx._context_handle, tld.device_name, âGreaterEqualâ, name,
â 4055 tld.op_callbacks, x, y)
4056 return _result
4057 except _core._NotOkStatusException as e:
TypeError: Cannot convert 0.5 to EagerTensor of dtype int64
What is wrong with the float number we are giving for the threshold?
Thanks