CV/PyTorch ResNet 使用与源码解析
本章代码:https://github.com/zhangxiann/PyTorch_Practice/blob/master/lesson8/resnet_inference.py
这篇文章首先会简单介绍一下 PyTorch
中提供的图像分类的网络,然后重点介绍 ResNet
的使用,以及 ResNet
的源码。
模型概览
在torchvision.model
中,有很多封装好的模型。
本章代码:https://github.com/zhangxiann/PyTorch_Practice/blob/master/lesson8/resnet_inference.py
这篇文章首先会简单介绍一下 PyTorch
中提供的图像分类的网络,然后重点介绍 ResNet
的使用,以及 ResNet
的源码。
在torchvision.model
中,有很多封装好的模型。