Added training for bounding box prediction.