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

Fixed SSD prediction

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-09 12:23:27 +02:00
parent 311be36acf
commit 01a09ba474
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -190,7 +190,7 @@ def predict_keras(generator: callable,
detections = None detections = None
batch_size = None batch_size = None
for _ in range(forward_passes_per_image): for _ in range(forward_passes_per_image):
predictions = ssd_model.model.predict_on_batch(x) predictions = ssd_model.predict_on_batch(x)
if batch_size is None: if batch_size is None:
batch_size = predictions.shape[0] batch_size = predictions.shape[0]
if detections is None: if detections is None:
@ -203,7 +203,7 @@ def predict_keras(generator: callable,
# do observation stuff # do observation stuff
predictions = np.asarray(_get_observations(detections)) predictions = np.asarray(_get_observations(detections))
else: else:
predictions = ssd_model.model.predict_on_batch(x) predictions = ssd_model.predict_on_batch(x)
decoded_predictions_batch = ssd_output_decoder.decode_detections_fast( decoded_predictions_batch = ssd_output_decoder.decode_detections_fast(
y_pred=predictions, y_pred=predictions,