Ghulam Jilani Raza comments

Results 4 comments of


                                            Ghulam Jilani Raza

How To Add NMS in pytorch model so that it gets converted into TFLITE

JFYI, I've added the TFlite custom post processing OP in a keras object detection model following [this](https://github.com/tensorflow/models/blob/master/research/object_detection/export_tflite_graph_lib_tf2.py) method, but since pytorch has no NMS layer, so its getting tricky here....

How To Add NMS in pytorch model so that it gets converted into TFLITE

> @gj-raza Is [`torchvision.ops.nms`](https://pytorch.org/vision/stable/ops.html#torchvision.ops.nms) sufficient for your usage? i've tried this but it seems to not work. have a look at below code ``` # Simple Pytorch model class PyModel(nn.Module):...

How To Add NMS in pytorch model so that it gets converted into TFLITE

> @gj-raza I've done the first two tasks. If you are interested, you may take a look at the latter two, which should be fairly easy. @peterjc123 Sure, but I...

How To Add NMS in pytorch model so that it gets converted into TFLITE

> @gj-raza The schema of the `TFLITE_DETECTION_POSTPROCESS` op is added [here](https://github.com/alibaba/TinyNeuralNetwork/commit/bd70e36e6cfa301c605ae3bc42db4eda159a183d#diff-4d917771c0f7ad00a838050a2ecc673f4ebadd28ec480330a5e4a1b905eb9c36R12) in case you may need that. Thanks @peterjc123 . So now i'll have to map this operator to a...