torch
关注
分享
扫一扫
gggi520
这个作者很懒，什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
ROIAlign python实现
【代码】ROIAlign python实现。原创 2023-01-16 16:03:07 · 572 阅读 · 0 评论 -
torch在vgg16预训练模型上添加新的层
import torchfrom torch import nnimport torchvisionvgg16=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 · 3443 阅读 · 1 评论