Added debug print for shape of true positives

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-07-18 14:53:44 +02:00
parent 797323f6e9
commit 60fb13cddf

View File

@ -324,6 +324,7 @@ def get_precision_recall(number_gt_per_class: np.ndarray,
diff_to_largest_class = cumulative_precision_micro.shape[0] - cumulative_precision.shape[0] diff_to_largest_class = cumulative_precision_micro.shape[0] - cumulative_precision.shape[0]
if diff_to_largest_class: if diff_to_largest_class:
print(f"Diff to largest: {diff_to_largest_class}") print(f"Diff to largest: {diff_to_largest_class}")
print(f"Shape: {tp.shape}")
repeated_last_precision = np.tile(cumulative_precision[-1], diff_to_largest_class) repeated_last_precision = np.tile(cumulative_precision[-1], diff_to_largest_class)
repeated_last_recall = np.tile(cumulative_recall[-1], diff_to_largest_class) repeated_last_recall = np.tile(cumulative_recall[-1], diff_to_largest_class)
extended_precision = np.concatenate((cumulative_precision, repeated_last_precision)) extended_precision = np.concatenate((cumulative_precision, repeated_last_precision))