2021年4月2日
PyTorch——模型搭建——ResNet18（一）
摘要： 试一试 1 import torch 2 import torch.nn as nn 3 import torch.nn.functional as F 4 from torchsummary import summary 5 6 class ResBlock(nn.Module): 7 def _ 阅读全文
posted @ 2021-04-02 16:43 一剑光寒十四州 阅读(821) 评论(0) 推荐(0)
PyTorch——模型推断——单张推断OpenCV（二）
摘要： 1 import os 2 import torch 3 from torchvision import transforms 4 from data_pipe import get_data 5 from vgg import VGG_13 6 from resnet18 import ResNe 阅读全文
posted @ 2021-04-02 16:40 一剑光寒十四州 阅读(96) 评论(0) 推荐(0)