Bounding box detection
PyTorch