将matlab神经网络工具箱代码导出Resnet18
ResNet-18
是一个卷积神经网络,它在来自
ImageNet
数据库的超过一百万张图像上进行了训练。
结果,网络已经为各种图像学习了丰富的特征表示。
该网络可以将图像分为
1000
个对象类别,例如键盘、鼠标、铅笔和许多动物。
该网络的图像输入大小为
224×224×3。
用法
此存储库需要
MATLAB(R2018b
及更高版本)和深度学习工具箱。
该存储库提供三个功能:
resnet18Layers:使用
ResNet-18
的网络架构创建未经训练的网络要从头开始训练网络,请在
MATLAB
命令行中键入以下内容:
lgraph
=
resnet18Layers;
未经训练的网络作为
layerGraph
对象返回。
要构建适用于图像分类的经过训练的
ResNet-18
网络,请在
MATLAB
命令行中键入以下内容:
净
=
assembleResNet18;
经过训练的网络作为
DAGNetwork
对象返回。
要使用网络对图像进行分类:
img
=
imresize(imread("peppers.png"),[22
2021-07-09 09:20:01
76KB
系统开源
1