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

Added missing function argument

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-24 11:33:13 +02:00
parent 856aad8e21
commit 3330ce69e7
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/twomartens/masterthesis/cli.py
View File

@ -26,11 +26,18 @@ Functions:
prepare(...): prepares the SceneNet ground truth data
"""
import argparse
from typing import Callable, Union, Tuple, Sequence, Optional, Generator, List, Any, Dict
from typing import Any
from typing import Callable
from typing import Dict
from typing import Generator
from typing import List
from typing import Optional
from typing import Sequence
from typing import Tuple
from typing import Union
import math
import numpy as np
import tensorflow as tf
from twomartens.masterthesis import config as conf
@ -876,7 +883,8 @@ def _ssd_test_get_generators(args: argparse.Namespace,
batch_size,
image_size,
training=False, evaluation=True, augment=False,
debug=args.debug)
debug=args.debug,
predictor_sizes=predictor_sizes)
else:
generator, length, debug_generator = load_data_scenenet(file_names, instances, coco_path,
predictor_sizes=predictor_sizes,