Hey there,
if this is a mistake
The section of Inspect _prediction_heads contains the information of ConvolutionalBoxPredictor
, however, the assignment will be working with WeightSharedConvolutionalBoxPredictor
.
For ConvolutionalBoxPredictor
the _prediction_heads
is
self._prediction_heads = {
BOX_ENCODINGS: box_prediction_heads,
CLASS_PREDICTIONS_WITH_BACKGROUND: class_prediction_heads,
}
For WeightSharedConvolutionalBoxPredictor
which is the goal of this assignment the _prediction_heads
is
self._prediction_heads = {
CLASS_PREDICTIONS_WITH_BACKGROUND: class_prediction_head,
}
I am not sure if the following
self._prediction_heads.update(other_heads)
will bring BOX_ENCODINGS
from outside.
PS//These two predictors are siblings, derived from KerasBoxPredictor
.
Thanks