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

Improved type hinting for image loading function

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-13 11:41:49 +02:00
parent 2beb904792
commit c1e82139a7
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/twomartens/masterthesis/data.py
View File

@ -303,8 +303,8 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]],
def _load_images_ssd_callback(resized_shape: Sequence[int]) \
-> Callable[[Sequence[str]],
Tuple[tf.Tensor]]:
-> Callable[[Sequence[str], Sequence[Sequence[int]]],
Tuple[tf.Tensor, Sequence[Sequence[int]]]]:
"""
Returns the callback function to load images for SSD.
@ -315,7 +315,8 @@ def _load_images_ssd_callback(resized_shape: Sequence[int]) \
callback function
"""
def _load_images_ssd(paths: Sequence[str], labels: Sequence[Sequence[int]]) -> Tuple[tf.Tensor]:
def _load_images_ssd(paths: Sequence[str],
labels: Sequence[Sequence[int]]) -> Tuple[tf.Tensor, Sequence[Sequence[int]]]:
"""
Callback function to load images for SSD.