import tensorflow as tf
修改为:
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
对于scipy,读取保存文件的方式:
from scipy.misc import imread, imsave
可以修改为:
import imageio.v2 as imageio
或使用opencv-python:
import cv2
对于tf.contrib.layers.batch_norm报错可以修改为:
tf.layers.batch_normalization()
传入参数设置为符合语法要求的形式,保持原设置。
with tf.variable_scope(scope_name):
计算flops和params通过加载计算图,使用profiler得到结果