pytorch
do_best_
这个作者很懒,什么都没留下…
展开
-
pytorch之transforms
pytorch torchvision transform对PIL.Image进行变换class torchvision.transforms.Compose(transforms)将多个transform组合起来使用。transforms: 由transform构成的列表. 例子:transforms.Compose([ transforms.CenterCrop...原创 2019-04-02 10:33:35 · 980 阅读 · 0 评论 -
pytorch之models
torchvision.models模块的子模块中包含以下模型结构。AlexNet VGG ResNet SqueezeNet DenseNet可以使用随机初始化的权重来创建这些模型。import torchvision.models as modelsresnet18 = models.resnet18()alexnet = models.alexnet()squeez...原创 2019-04-02 10:43:14 · 2071 阅读 · 0 评论