语义分割
王吉吉丫
这个作者很懒,什么都没留下…
展开
-
Pytorch框架下的语义分割实战(三,网络训练)
接上一篇。。。今天我们看一看Z博是怎么训练网络的。。。附上代码'''train.py'''from datetime import datetimeimport numpy as npimport torchimport torch.nn as nnimport torch.optim as optimfrom dataset import test_dataloader, train_dataloaderfrom model import FCNs,VGGNet#我将原创 2020-10-16 08:44:18 · 1562 阅读 · 2 评论 -
Pytorch框架下的语义分割实战(二,网络搭建)
接上一篇Pytorch框架下的语义分割实战(数据集处理)我们接着聊聊FCN.py中的小细节的吧。。。先附上代码# -*- coding: utf-8 -*-"""FCN.py"""import torch.nn as nnfrom torchvision.models.vgg import VGG#继承nn.Module,撰写自己的网络层class FCNs(nn.Module): ''' 类FCNs:将最后一个特征图直接上采样32倍(5次步长为2、卷积核为原创 2020-10-13 09:51:12 · 934 阅读 · 0 评论