fix mixed precision eval when use COCOCallback

Author: fsx950223

efficientdet/keras/efficientdet_keras.py

@@ -940,10 +940,6 @@ def _postprocess(self, cls_outputs, box_outputs, scales, mode='global'):
     if not mode:
       return cls_outputs, box_outputs
 
-    # TODO(tanmingxing): remove this cast once FP16 works postprocessing.
-    cls_outputs = [tf.cast(i, tf.float32) for i in cls_outputs]
-    box_outputs = [tf.cast(i, tf.float32) for i in box_outputs]
-
     if mode == 'global':
       return postprocess.postprocess_global(self.config.as_dict(), cls_outputs,
                                             box_outputs, scales)

efficientdet/keras/postprocess.py

@@ -28,9 +28,9 @@
 
 def to_list(inputs):
   if isinstance(inputs, dict):
-    return [inputs[k] for k in sorted(inputs.keys())]
+    return [tf.cast(inputs[k], tf.float32) for k in sorted(inputs.keys())]
   if isinstance(inputs, list):
-    return inputs
+    return [tf.cast(i, tf.float32) for i in inputs]
   raise ValueError('Unrecognized inputs : {}'.format(inputs))
 
 

efficientdet/requirements.txt

@@ -7,6 +7,7 @@ PyYAML>=5.1
 six>=1.15.0
 tensorflow>=2.4.0
 tensorflow-addons>=0.12
+tensorflow-hub>=0.11
 neural-structured-learning>=1.3.1
 tensorflow-model-optimization>=0.5
 Cython>=0.29.13