From 9ee069a4f15edc7568faf70cd76ad3606e43114b Mon Sep 17 00:00:00 2001 From: HaydenFaulkner Date: Mon, 30 Dec 2019 11:38:10 +1030 Subject: [PATCH] add FLAGS for temp model to testing script --- detect_yolo3.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/detect_yolo3.py b/detect_yolo3.py index 6c6a155..363199d 100644 --- a/detect_yolo3.py +++ b/detect_yolo3.py @@ -85,6 +85,10 @@ 'Type to join hierarchical darknet. can be max or conv.') flags.DEFINE_list('hier', [1, 1, 1, 1, 1], 'the hierarchical factors, the input must be temporally equal to all these multiplied together') +flags.DEFINE_boolean('mult_out', False, + 'Have one or multiple outs for timeseries data') +flags.DEFINE_boolean('temp', False, + 'Use new temporal model') flags.DEFINE_boolean('visualise', False, 'Do you want to display the detections?') @@ -632,7 +636,7 @@ def main(_argv): block_conv_type=FLAGS.block_conv_type, rnn_pos=FLAGS.rnn_pos, corr_pos=FLAGS.corr_pos, corr_d=FLAGS.corr_d, motion_stream=FLAGS.motion_stream, agnostic=FLAGS.model_agnostic, add_type=FLAGS.stream_gating, new_model=FLAGS.new_model, - hierarchical=FLAGS.hier, h_join_type=FLAGS.h_join_type) + hierarchical=FLAGS.hier, h_join_type=FLAGS.h_join_type, temporal=FLAGS.temp, t_out=FLAGS.mult_out) else: net = yolo3_3ddarknet(trained_on_dataset.classes, conv_types=FLAGS.conv_types) else: