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

Corrected type hinting

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-03-21 17:31:42 +01:00
parent 976563526a
commit 69cc07ed12
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -38,8 +38,11 @@ import os
import pickle import pickle
import time import time
from typing import Dict from typing import Dict
from typing import List
from typing import Optional from typing import Optional
from typing import Union
import numpy as np
import tensorflow as tf import tensorflow as tf
from tensorflow.python.ops import summary_ops_v2 from tensorflow.python.ops import summary_ops_v2
@ -168,7 +171,7 @@ def predict(dataset: tf.data.Dataset,
def _predict_one_epoch(dataset: tf.data.Dataset, def _predict_one_epoch(dataset: tf.data.Dataset,
use_dropout: bool, use_dropout: bool,
forward_passes_per_image: int, forward_passes_per_image: int,
ssd: tf.keras.Model) -> Dict[str, float]: ssd: tf.keras.Model) -> Dict[str, Union[float, List[List[np.ndarray]]]]:
with summary_ops_v2.always_record_summaries(): with summary_ops_v2.always_record_summaries():
epoch_start_time = time.time() epoch_start_time = time.time()