本博客通过tensorflow实现resnet18和resnet34模型,对于resnet模型有不同的写法,包括,sequence模型、类封装、自定义函数,而本博客主要用自定义函数和类封装实现resnet18和34,代码和图如下:
![请添加图片描述](https://img-blog.csdnimg.cn/6c40e398881a444e969c35c244e658f7.png)
本博客主要用的cifar10数据集,图片大小是32x32x3,训练模型精度不是很好,有待改进一下继续提高精度。流程图如下:
![在这里插入图片描述](https://img-blog.csdnimg.cn/307c58859b3c47e08203345609e7074d.jpeg#pic_center)
1. resnet-18
import tensorflow as tf
from tensorflow.keras.layers import Conv2D,MaxPooling2D,Flatten