![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
愚石lei
这个作者很懒,什么都没留下…
展开
-
Pytorch查看网络结构
1. Thop1.1 安装打开cmd,输入pip install thop。(链接Thop)1.2 使用from torchvision.models import resnet50 from thop import profile model = resnet50() input = torch.randn(1, 3, 224, 224) flops, params...原创 2019-11-20 10:07:18 · 8700 阅读 · 1 评论 -
Pytorch——ImageFolder使用
ImageFolder是一个通用的数据加载器,数据集应当按照指定的格式进行存储。1 数据集构造方式比如我们的数据集一共包括两个类别:cat、dog,每个类别包括四张图片。所有的图片按文件夹保存,每个文件夹下存储同一个类别的图片,文件夹名为类名。2 ImageFolder2.1 调用import torchvision.datasetsdataset=ImageFo...原创 2019-07-02 11:17:44 · 8574 阅读 · 0 评论