基于Android pytorch API 深度学习语义分割（下）

落日映苍穹つ 2022-09-11 10:20 237阅读 0赞

草稿

import numpy as np
    import torch.nn.functional as F
    import cv2
    from dataset import Mydataset
    from fcNet import *
    from torch.utils.data import DataLoader
    import torch.nn as nn
    import torch
    from eval import *
    
    numclass = 2
    epoches = 100
    train_data = Mydataset(352, 480, True)
    train_loder = DataLoader(train_data, batch_size=8, shuffle=True, num_workers=0)
    val_data = Mydataset(352, 480, False)
    val_loder = DataLoader(train_data, batch_size=8, shuffle=True, num_workers=0)
    fcn=MyFcNet_vgg16bn(2)
    cretion = nn.NLLLoss()
    optimizer = torch.optim.Adam(fcn.parameters(), lr=1e-4)
    
    
    
    def train(model):
        best=[0]
        net = model.train()
    
        for epoch in range(epoches):
            print("Epoch is [{}/{}]".format(epoch + 1, epoches))
            if epoch % 50 == 0 and epoch != 0:
                for gp in optimizer.param_groups:
                    gp['lr'] *= 0.5
    
            train_loss =

发表评论取消回复

表情：

评论列表（有 0 条评论，237人围观）

还没有评论，来说两句吧...

相关阅读

相关基于深度学习的图像语义分割（Deep Learning-based Image Semantic Segmentation）

![2bd9eec337f547eda641893aafeb5718.png][] 目录编辑深度学习算法中的基于深度学习的图像语义分割深度学习的图像语义分割算法

浅浅的花香味﹌/ 2024年02月28日 08:00/ 0 赞/ 61 阅读

相关用于语义分割问题的深度学习技术综述

近期主要在学习语义分割相关方法，计划将arXiv上的这篇综述好好翻译下，目前已完成了一部分，但仅仅是尊重原文的直译，后续将继续完成剩余的部分，并对文中提及的多个方法给出自己的理

谁借莪１个温暖的怀抱￠/ 2024年01月20日 08:00/ 0 赞/ 56 阅读

相关 Pytorch 深度学习实战教程（二）：UNet语义分割网络

![Pytorch深度学习实战教程（二）：UNet语义分割网络][Pytorch_UNet] > 本文 GitHub [https://github.com/Jack-Che

Bertha 。/ 2023年02月11日 10:23/ 0 赞/ 231 阅读

相关 Pytorch深度学习实战教程（一）：开启炼丹模式，语义分割基础与环境搭建

> 本文 GitHub [https://github.com/Jack-Cherish/PythonPark][https_github.com_Jack-Cherish_P

你的名字/ 2023年02月11日 07:15/ 0 赞/ 14 阅读

相关 HALCON 20.11：深度学习笔记(12)---语义分割

HALCON 20.11：深度学习笔记(12)--- 语义分割 HALCON 20.11.0.0中，实现了深度学习方法。本章解释了如何使用基于深度学习的语义分割，包括

╰+哭是因爲堅強的太久メ/ 2022年12月30日 08:27/ 0 赞/ 573 阅读

相关【深度学习】Transformer在语义分割上的应用探索

【深度学习】Transformer在语义分割上的应用探索文章目录 1 Segmenter 2 Swin-Unet：Unet形状的纯Transfo

不念不忘少年蓝@/ 2022年10月10日 01:37/ 0 赞/ 203 阅读

相关语义分割学习笔记3——运行语义分割程序

语义分割学习笔记3——运行语义分割程序一、linux安装依赖环境 1、创建open-mmlab环境 2、安装pytorch

清疚/ 2022年10月08日 11:16/ 0 赞/ 626 阅读

相关语义分割学习笔记2——Jetson Xavier安装pytorch

语义分割学习笔记2——Jetson Xavier安装pytorch 一、whl文件安装pytorch 二、安装torchvision 三、测试是否安装成

╰+攻爆jí腚メ/ 2022年10月05日 05:44/ 0 赞/ 126 阅读

相关基于Android pytorch API 深度学习语义分割（下）

草稿 import numpy as np import torch.nn.functional as F import cv2 from d

落日映苍穹つ/ 2022年09月11日 10:20/ 0 赞/ 238 阅读

相关毕业设计之 - 基于深度学习的图像语义分割

1 简介 Hi，大家好，这里是丹成学长，今天向大家介绍基于深度学习的图像语义分割大家可用于毕业设计技术解答、开题指导、毕设帮助毕设帮助：<

柔光的暖阳◎/ 2022年09月10日 07:17/ 0 赞/ 452 阅读