PyTorch object detection with pre-trained networks