pytorch
smallworldxyl
这个作者很懒,什么都没留下…
展开
-
MobileNetV3代码(pytorch)
MobileNetV3-Large结构图第一列Input代表mobilenetV3每个特征层的shape变化;第二列Operator代表每次特征层即将经历的block结构,我们可以看到在MobileNetV3中,特征提取经过了许多的bneck结构;第三、四列分别代表了bneck内逆残差结构上升后的通道数、输入到bneck时特征层的通道数。第五列SE代表了是否在这一层引入注意力机制。第六列NL代表了激活函数的种类,HS代表h-swish,RE代表RELU。第七列s代表了每一次block结构所用的步原创 2021-11-28 21:02:40 · 6678 阅读 · 6 评论 -
Unet++网络结构代码(pytorch)
网络结构图如下:import torchfrom torch import nnimport torch.nn.functional as Ffrom torch.utils.tensorboard import SummaryWriterclass VGGBlock(nn.Module): def __init__(self, in_channels, middle_channels, out_channels): super().__init__()原创 2021-11-28 14:33:33 · 10468 阅读 · 2 评论 -
pytorch完整套路模板代码(CIFAR10数据集为例)
import torchimport torchvisionfrom torch.utils.tensorboard import SummaryWriter# from model import *# 准备数据集from torch import nnfrom torch.utils.data import DataLoadertrain_data = torchvision.datasets.CIFAR10(root="./data", train=True, transform=to原创 2021-11-18 17:14:10 · 226 阅读 · 0 评论 -
pytorch编写自己的dataset
from torch.utils.data import Dataset, DataLoaderfrom PIL import Imageimport osclass MyData(Dataset): def __init__(self, root_dir, label_dir): self.root_dir = root_dir self.label_dir = label_dir self.path = os.path.join(..原创 2021-11-16 10:54:16 · 1778 阅读 · 1 评论