hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
https://github.com/hunglc007/tensorflow-yolov4-tflite
MIT License
2.25k stars 1.24k forks source link

Output probabilities for all the classes (for each bounding box)/ detect only in specific areas of the video #186

Open hirsiAbdollahi opened 4 years ago

hirsiAbdollahi commented 4 years ago

Hi guys! have to 2 questions:

Desirce Output:

[ {'object_position' : {'bx': 0.8755, 'by':'0.12, 'bh': 0.2, 'bw' : 0.5}, 'object_prediction' :{'house' : 0.24, 'boat':0.65, ...} },

{ {'object_position' : {'bx': 0.565, 'by':'0.2472, 'bh': 0.1, 'bw' : 0.3}, 'object_prediction' :{'house' : 0:04, 'boat':0.355, ...} }

]

Thanks in advance