Implement a YOLO (v3) object detector from scratch in PyTorch