Internal change

PiperOrigin-RevId: 519275194

Author: tensorflower-gardener

official/legacy/detection/README.md

@@ -43,7 +43,7 @@ The pre-trained ResNet-50 checkpoint can be downloaded [here](https://storage.cl
 Note: The ResNet implementation under
 [detection/](https://github.com/tensorflow/models/tree/master/official/legacy/detection)
 is currently different from the one under
-[classification/](https://github.com/tensorflow/models/blob/master/docs/vision/image_classification.ipynb),
+[classification/](https://github.com/tensorflow/models/tree/master/official/vision/image_classification),
 so the checkpoints are not compatible.
 We will unify the implementation soon.
 

official/vision/evaluation/coco_evaluator.py

@@ -287,7 +287,8 @@ def _process_bbox_predictions(self, predictions):
       predictions['detection_outer_boxes'] /= image_scale
 
   def _process_keypoints_predictions(self, predictions):
-    image_scale = predictions['image_info'][:, 2:3, :]
+    image_scale = tf.reshape(predictions['image_info'][:, 2:3, :],
+                             [-1, 1, 1, 2])
     predictions['detection_keypoints'] = (
         predictions['detection_keypoints'].astype(np.float32))
     predictions['detection_keypoints'] /= image_scale