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

Added debug prints to get shapes 

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-07-15 10:46:15 +02:00
parent c0d23a731b
commit 2ecc0023f8
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -340,6 +340,10 @@ def _predict_save_images(inputs: np.ndarray, predictions: np.ndarray,
get_coco_cat_maps_func: callable,
output_path: str, coco_path: str,
image_size: int) -> None:
print(
f"Input shape: {inputs.shape}"
f"Predictions shape: {predictions.shape}"
)
save_images(inputs, predictions,
output_path, coco_path, image_size,
get_coco_cat_maps_func, "after-prediction")