Yolo non max suppression
I am getting that yolo_filter_boxes is correct, function iou is correct but yolo_non_max_suppression is throwing the following error:
InvalidArgumentError: Value for attr ‘T_threshold’ of int32 is not in the list of allowed values: half, float
; NodeDef: {{node NonMaxSuppressionV3}}; Op<name=NonMaxSuppressionV3; signature=boxes:T, scores:T, max_output_size:int32, iou_threshold:T_threshold, score_threshold:T_threshold → selected_indices:int32; attr=T:type,default=DT_FLOAT,allowed=[DT_HALF, DT_FLOAT]; attr=T_threshold:type,default=DT_FLOAT,allowed=[DT_HALF, DT_FLOAT]> [Op:NonMaxSuppressionV3]
Yes working in Coursera Labs, I use max_boxes_tensor variable when I call non_max_suppression but have also tried it with max_boxes and with max_output_size=max_boxes
Thanks for your suggestion as to the nature of the bug here, but please note that this thread is more than 6 months old. It’s pretty likely that the author of the original thread has found a solution. Of course your comment could still benefit someone new who finds this thread through the forum search.