本博客通过tensorflow实现resnet18和resnet34模型,对于resnet模型有不同的写法,包括,sequence模型、类封装、自定义函数,而本博客主要用自定义函数和类封装实现resnet18和34,代码和图如下:
本博客主要用的cifar10数据集,图片大小是32x32x3,训练模型精度不是很好,有待改进一下继续提高精度。流程图如下:
1. resnet-18
import tensorflow as tf
from tensorflow.keras.layers import Conv2D,MaxPooling2D,Flatten
版权声明:本文为m0_47405013原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。