search

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

Update detect.py #262

Open jjerry-k opened 3 years ago

jjerry-k commented 3 years ago

value's shape: [batch, h, w, 3, 5+num_class] [..., :4] and [..., 4:] is more efficient than [:, :, 0:4] and [:, :, 4:]