Commit 86b4eca8 authored by cq615's avatar cq615

Update main.py

parent fa7517a8
......@@ -30,7 +30,7 @@ if __name__ == '__main__':
label_var = T.itensor4('label')
image_seg_var = T.tensor4('seg')
# Compile the model
# Compile the model (CNN one, to build RNN one, compile build_FCN_triple_branch_rnn and use sequence data)
net = build_FCN_triple_branch(image_var, image_pred_var, image_seg_var, shape=(None, 1, size, size))
model_file = os.path.join(model_path, 'FCN_VGG16_sz192_joint_learn_ft_tmp.npz')
with np.load(model_file) as f:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment