tf2.0没有fully_connected()全连接层函数
今天在做卷积层的时候,要做一个全连接层,发现调用 tf.contrib.layers.fully_connected()调用失败,查阅发现tf2.0已经去除了contrib的库,因为该库比较繁琐,不容易维护,所以已经逐渐去除了,contrib库中的layers但是还是保存的,所以有些函数还是存在的,直接调用 tf.compat.v1.layers.有些函数还是存在的,但是fully_connected()已经去除。
显然这里再去想方设法的调用contrib库已经不可能了,有两种方法去代替,第一种就是用tf2.0 新的全连接层dense来代替
下面文章
Tensorflow2.0 全连接层
也可以直接用 tf.compat.v1.layers.dense() 参数还是和fully_connected中的差不多,这样可以代替fully_connected
但是还是有强迫症,还是想用原来的fully_connected,其实在contrib库去除中,将其中的函数整合到了tf_slim库中,这个库是tf的瘦身版,所以叫tf_slim 它直接包含了tf的一些函数,可以直接调用,其中也就包括fully_connected, 怎么下载?一般使用pycharm 时候大家都习惯使用在setting中搜索下载,但是如果没有,那么就在pycharm下面的Terminal端代码下载:
输入pip install tf_lism
那么通过Terminal端下载的包在哪?你在setting中搜索是找不到的,一般下载到了项目的外部资源中了:
下拉你就会找到tf_slim包,那么直接就可以import
import tf_slim
它的过人之处就是不用输一堆库再调用函数,直接
tf_slim.fully_connected()
就可以,因为它将大部分的tf函数都整合到一块了,没有分类,想调用那个直接调用tf_slim即可