2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed transformation of predictions

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-11 17:33:08 +02:00
parent 38229e1848
commit ba3cfa379e
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/twomartens/masterthesis/ssd.py
View File

@ -301,7 +301,8 @@ def _transform_predictions(predictions: np.ndarray, inverse_transforms: Sequence
decoded_predictions = decode_func( decoded_predictions = decode_func(
y_pred=predictions, y_pred=predictions,
img_width=image_size, img_width=image_size,
img_height=image_size img_height=image_size,
coords="corners"
) )
transformed_predictions = inverse_transform_func(decoded_predictions, inverse_transforms) transformed_predictions = inverse_transform_func(decoded_predictions, inverse_transforms)