Implementation of YOLO object detection