Tensorfow2.0不能使用get_session问题

RuntimeError: `get_session` is not available when using TensorFlow 2.0.

yolov3中源代码有这样:

self.sess = K.get_session()

报错显示tf2.0以上移除了这函数,网上大部分方法都是降低tf版本,参考了一篇博文,解决如下:

将self.sess = K.get_session()改为

self.sess = tf.compat.v1.Session()

 

你可能感兴趣的:(Tensorfow2.0不能使用get_session问题)