Skip to content

Commit 0177dee

Browse files
YknZhuaquariusjay
authored andcommitted
Fix export model script (#7791)
1 parent 87bc52c commit 0177dee

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

research/deeplab/export_model.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,15 +17,14 @@
1717

1818
import os
1919
import tensorflow as tf
20-
from google3.learning.brain.contrib import quantize as contrib_quantize
21-
from google3.learning.brain.contrib import slim as contrib_slim
2220

21+
from tensorflow.contrib import quantize as contrib_quantize
2322
from tensorflow.python.tools import freeze_graph
2423
from deeplab import common
2524
from deeplab import input_preprocess
2625
from deeplab import model
2726

28-
slim = contrib_slim
27+
slim = tf.contrib.slim
2928
flags = tf.app.flags
3029

3130
FLAGS = flags.FLAGS

0 commit comments

Comments
 (0)