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

Added further debug prints

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-30 14:20:55 +02:00
parent 6eb81c4568
commit 33635b5c49
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -262,11 +262,13 @@ def _predict_one_epoch(dataset: tf.data.Dataset,
def _get_observations(detections: Sequence[Sequence[np.ndarray]]) -> List[List[np.ndarray]]: def _get_observations(detections: Sequence[Sequence[np.ndarray]]) -> List[List[np.ndarray]]:
batch_size = len(detections) batch_size = len(detections)
observations = [[] for _ in range(batch_size)] observations = [[] for _ in range(batch_size)]
print(f"batch size: {batch_size}")
# iterate over images # iterate over images
for i in range(batch_size): for i in range(batch_size):
print(f"{i}th batch element") print(f"{i}th batch element")
detections_image = np.asarray(detections[i]) detections_image = np.asarray(detections[i])
print(detections_image.shape)
overlaps = bounding_box_utils.iou(detections_image[:, -12:-8], overlaps = bounding_box_utils.iou(detections_image[:, -12:-8],
detections_image[:, -12:-8], detections_image[:, -12:-8],
mode="outer_product", mode="outer_product",