def _setup_graph(self):
self._init_mask_op = tf.initialize_variables(tf.get_collection('masks'))
self._init_thre_op = tf.initialize_variables(tf.get_collection('thresholds'))
微信
公众号
扫码关注公众号