def __init__(self, FLAGS, sess): # init model self.sess = sess self.density_model = get_network("density") self.flags = FLAGS self.writer = SummaryWriter() self.frame_shape = (FLAGS.img_height, FLAGS.img_width)
def __init__(self, sess): self.sess = sess self.density_model = get_network("density")