Tensorflow2.0学习(七):猫狗大战2、训练与保存模型一、搭建网络二、读取tfrecord数据三、训练与保存模型的两种方式四、预测一张图像,并输出概率值
熟悉TF1.X,对Record数据情有独钟,那么TF2.0如何训练record数据呢?在Tensorflow2.0学习(六):猫狗大战1、制作与读取record数据 已经介绍了如何生成与读取tfRecord数据,这篇介绍训练record数据与保存模型的两种方法.一、搭建网络以ResNet50为例,通常去掉最后一层的网络,最为新任务的特征提取子网络,然后根据自己的任务类别,加一个对应数据的全连...