使用PyTorch的ResNet网络进行图像分类
这个文件包含了ResNet18、ResNet50和ResNet101网络在PyTorch框架下实现图像分类的代码。我们还提供了单一测试和批量化测试代码来验证您的模型。ResNet网络是在VGG19的基础上通过短路机制加入残差单元来进行优化的。使用PyTorch框架可以使您轻松地拿走并且简单地使用这个分类器来对您的图像进行分类。
下载地址
用户评论
