mystic123 / tensorflow-yolo-v3

Implementation of YOLO v3 object detector in Tensorflow (TF-Slim)
https://medium.com/@pawekapica_31302/implementing-yolo-v3-in-tensorflow-tf-slim-c3c55ff59dbe
Apache License 2.0
893 stars 353 forks source link

how to get the predictions from the output_details #99

Open azizHakim opened 4 years ago

azizHakim commented 4 years ago

After converting the yolov3.eights into tflite format I am getting the output_details looks like this:

[{'name': 'Identity', 'index': 0, 'shape': array([ 1, 22743, 4], dtype=int32), 'dtype': <class 'numpy.float32'>, 'quantization': (0.0, 0)}, {'name': 'Identity_1', 'index': 1, 'shape': array([ 1, 22743, 80], dtype=int32), 'dtype': <class 'numpy.float32'>, 'quantization': (0.0, 0)}]

It seems like I am getting 2 output lists with shape [1, 22743, 4] and [1, 22743, 80] How do I get bounding boxes and labels from these outputs. Also, is it the proper output shape? I was expecting 4 output lists, bbox, labels, scores, and number of bbox.