![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
torch
gggi520
这个作者很懒,什么都没留下…
展开
-
ROIAlign python实现
【代码】ROIAlign python实现。原创 2023-01-16 16:03:07 · 399 阅读 · 0 评论 -
torch在vgg16预训练模型上添加新的层
import torch from torch import nn import torchvision vgg16=torchvision.models.vgg16(pretrained=False) train_data=torchvision.datasets.CIFAR10('./data',train=True, transform=torchvision.transforms.ToTensor(),download=True) print(vgg16) 此时输出vgg16结构 VGG(原创 2022-04-04 11:09:25 · 3185 阅读 · 1 评论