Creating training set for Yolo object detection