Describe the feature and the current behavior/state.
The fast-nms algorithm has better performance than traditional nms.
Described in paper.
Kernel implementation may provides better performance than python implementation.
Relevant information
Which API type would this fall under (layer, metric, optimizer, etc.)
Who will benefit with this feature?
Any other info.
Tensorflow Implementation on colab
Would like to see how fast it is compared with tf.image.non_max_suppression.
Is anyone working on this? If not, I will take a look @seanpmorgan
Is anyone working on this? If not, I will take a look @seanpmorgan
Not that I'm aware of, but @fsx950223 said they are willing to implement. If there has been no progress then should be fine to start on.
I'm a little busy for now, I will try it next week. Thanks.
I will wait for the next update on this.
@fsx950223 how does this algorithm compare against existing GPU NMS implementations in TF? Do you have any idea?
@fsx950223 how does this algorithm compare against existing GPU NMS implementations in TF? Do you have any idea?
Tensorflow benchmark could compare performance between them.
We have an upstream fast nms kernel in Tensorflow Lite https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/kernels/detection_postprocess.cc#L456-L463. @seanpmorgan I think this could be more in the "perimeter" of keras-cv (/cc @tanzhenyu ) as yesterday we have closed the other image processing related issues/PRs.