def nms_bboxes(scores_pred, bboxes_pred, nms_topk, nms_threshold, name):
with tf.name_scope(name, 'nms_bboxes', [scores_pred, bboxes_pred]):
idxes = tf.image.non_max_suppression(bboxes_pred, scores_pred, nms_topk, nms_threshold)
return tf.gather(scores_pred, idxes), tf.gather(bboxes_pred, idxes)
想把scores_pred和bboxes_pred轉換成NDArray,用eval()總是出各種錯誤,該怎么弄?我用的是Tensorflow1.5
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/24777.html
標籤:機器視覺
上一篇:軟體合同沒有明確驗收的細節需求及驗收標準怎么處理的?
下一篇:請教黑白圖的問題
