def init_network(self):
# jh-future:sizes can be add into tf.app.flags
network = my_model.MyInception(FLAGS.num_classes - FLAGS.labels_offset,
[FLAGS.scale_size1, FLAGS.scale_size2],
FLAGS.model_name,
is_training=False)
self.network = network
def init_network(self):
# jh-future:sizes can be add into tf.app.flags
network = my_model.MyInception(
FLAGS.num_classes - FLAGS.labels_offset,
[FLAGS.size_branch_0, FLAGS.size_branch_1],
FLAGS.model_name,
is_training=True)
self.network = network