Object Detection in PyTorch