saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-28 20:11:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/contrib_ops
History
Hector Li b1115f49cd
Update NMS to compatible with both TF & Pytoch models (#636) 
* Update NMS to compatible with both TF & Pytoch models

* update text

* set max_output_boxes_per_batch, iou_threshold, score_threshold as optional input to support dynamic value

* fix typo

* Set the last output selected_indices as optional output

* fix shape inference in case the input don't have shape

* Update schema to remove scores & boxes from output. support for class broadcast.

* change max_output_boxes_per_batch to max_output_boxes_per_class

* update schema to remove the class dimension from boxes
2019-04-01 16:08:02 -07:00
..
cpu Update NMS to compatible with both TF & Pytoch models (#636) 2019-04-01 16:08:02 -07:00
contrib_kernels.cc Update NMS to compatible with both TF & Pytoch models (#636) 2019-04-01 16:08:02 -07:00
contrib_kernels.h Kezhan/kernel registry refine (#346) 2019-01-18 09:55:30 -08:00