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

Added steps to arguments for SSDInputEncoder

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-05 12:48:19 +02:00
parent efda35adb7
commit 9f02297a52
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -403,7 +403,8 @@ def train(dataset: tf.data.Dataset,
# input encoder
input_encoder = ssd_input_encoder.SSDInputEncoder(IMAGE_SIZE[0], IMAGE_SIZE[1],
N_CLASSES, ssd.predictor_sizes)
N_CLASSES, ssd.predictor_sizes,
steps=[8, 16, 32, 64, 100, 300])
def _get_last_epoch(epoch_var: tf.Variable, **kwargs) -> int:
return int(epoch_var)