pytorch
ErrorMaker...
这个作者很懒,什么都没留下…
展开
-
google colab的数据加载和存储
1.启动云盘,即可看到如下from google.colab import drivedrive.mount('/content/drive')调整到当前目录 %cd /content/drive/My Drive!ls可以指从本地上传和下载,也可从网上直接下载!wget https://bin.equinox.io/c/4VmDzA7iaHb/ngrok-stable-linux-amd64.zip!unzip ngrok-stable-linux-amd64.zip原创 2021-09-29 20:03:22 · 372 阅读 · 0 评论 -
model参数的存储和调用
# 模型的存储torch.save(xxx_model.state_dict(),'./model_param.pth')model = Net().to(device)model.load_state_dict(torch.load('./model_param.pth'))原创 2021-09-28 22:31:18 · 287 阅读 · 0 评论 -
resnet18的模型使用
输入图片格式为:(b,c,224,224)修改输出层的输出维度:import torchvisionresnet_model = torchvision.models.resnet18(pretrained=True)for param in resnet_model.parameters(): param.requires_grad = Falseresnet_model.fcclass Net(nn.Module): def __init__(self, mo原创 2021-09-28 22:14:42 · 1427 阅读 · 0 评论 -
图片数据的读取
#训练数据增强,测试和验证数据只需转化为tensortrain_transform = transforms.Compose([transforms.Resize((224,224)), transforms.RandomHorizontalFlip(),transforms.ToTensor(),transforms.Normalize([0.485, 0.456, -.406],[0.229, 0.224, 0.225原创 2021-09-28 22:10:23 · 166 阅读 · 0 评论 -
torchvision.models.resnet18(pretrained=True)的直接用法
原创 2021-09-28 20:18:27 · 4810 阅读 · 0 评论