Skip to content

GitLab

Commit 0177deeb authored by Yukun Zhu's avatar Yukun Zhu Committed by aquariusjay
Browse files

Fix export model script (#7791)

parent 87bc52c5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
research/deeplab/export_model.py
View file @ 0177deeb
...@@ -17,15 +17,14 @@ ...@@ -17,15 +17,14 @@
import os import os
import tensorflow as tf import tensorflow as tf
from google3.learning.brain.contrib import quantize as contrib_quantize
from google3.learning.brain.contrib import slim as contrib_slim
from tensorflow.contrib import quantize as contrib_quantize
from tensorflow.python.tools import freeze_graph from tensorflow.python.tools import freeze_graph
from deeplab import common from deeplab import common
from deeplab import input_preprocess from deeplab import input_preprocess
from deeplab import model from deeplab import model
slim = contrib_slim slim = tf.contrib.slim
flags = tf.app.flags flags = tf.app.flags
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
......
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