Fixed transformation of predictions
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
parent
38229e1848
commit
ba3cfa379e
|
@ -301,7 +301,8 @@ def _transform_predictions(predictions: np.ndarray, inverse_transforms: Sequence
|
|||
decoded_predictions = decode_func(
|
||||
y_pred=predictions,
|
||||
img_width=image_size,
|
||||
img_height=image_size
|
||||
img_height=image_size,
|
||||
coords="corners"
|
||||
)
|
||||
transformed_predictions = inverse_transform_func(decoded_predictions, inverse_transforms)
|
||||
|
||||
|
|
Loading…
Reference in New Issue