一、源码准备

1.源码下载

可以直接到up主对应的github上进行下载，

GitHub - bubbliiiing/deeplabv3-plus-pytorch: 这是一个deeplabv3-plus-pytorch的源码，可以用于训练自己的模型。

他的代码里面进行了中文说明，对小白很友好

2.训练所需要的模型下载

链接: 百度网盘请输入提取码
提取码: 6n2c

下载完之后放至源码根目录中model_data文件夹，没有就新建一个

二、训练步骤

1.准备数据集

标注教程可看我上一篇文章分割数据集准备——labelme使用教程_道人兄的博客-CSDN博客

2.制作VOC数据集

① 完成步骤1之后，训练前将图片文件和.json放在dataste/before/文件夹下，然后修改json_to_dataset.py文件中的classes，保留background，添加自己的类型

之后便可以在JPEGImages中看到自己的图片，SegmentationClass看到制作好后的标注文件。

②将①图片文件放在VOCdevkit文件夹下的VOC2007文件夹下的JPEGImages中，将标签文件放在VOCdevkit文件夹下的VOC2007文件夹下的SegmentationClass中。运行根目录下的voc_annotation.py，从而生成train.txt和val.txt

3.修改参数

①train.py里的num_classes ，num_classes用于指向检测类别的个数+1！训练自己的数据集必须要修改！

②运行根目录下的train.py文件就可以开始训练啦~

有什么疑问或者建议评论区留言~

DeeplabV3+训练数据集流程（学习记录）相关推荐

SSD训练数据集流程（学习记录）
关于理论部分我看的是b站"霹雳吧啦Wz"的SSD理论讲解,作为入门小白表示能听懂,需要的同学可以自行观看目录 1.训练环境 2.训练步骤 1.训练环境我的环境是win11+an ...
Cadence Allegro 设计流程学习记录
Cadence Allegro 设计流程学习记录前提摘要软件设计版本: 电路仿真软件:NI Multisim 14.0,TINA-TI. 原理图设计:Design Entry CIS 16.6. ...
Android Camera 流程学习记录（五）—— Camera.takePicture() 流程解析
简介在前面的几篇笔记中,我已经把 Camera 控制流的部分梳理得比较清楚了.在 Camera 流程中,还有一个重要的部分,即数据流. Camera API 1 中,数据流主要是通过函数回调的方式, ...
BIM概述及应用流程学习记录
文章目录前言一.BIM是什么? BIM Revit : buiding information modeling 二.BIM概述及应用流程 1.什么是BIM 2.BIM如何来实施实施的步骤: 3 ...
【深度学习】深度学习模型训练全流程！
Datawhale干货作者:黄星源.奉现,Datawhale优秀学习者本文从构建数据验证集.模型训练.模型加载和模型调参四个部分对深度学习中模型训练的全流程进行讲解. 一个成熟合格的深度学习训练流 ...
加载tf模型正确率很低_深度学习模型训练全流程！
↑↑↑关注后"星标"Datawhale 每日干货 & 每月组队学习,不错过 Datawhale干货作者:黄星源.奉现,Datawhale优秀学习者本文从构建数据验证集. ...
深度学习模型训练全流程！
↑↑↑关注后"星标"Datawhale 每日干货 & 每月组队学习,不错过 Datawhale干货作者:黄星源.奉现,Datawhale优秀学习者本文从构建数据验证集. ...
基于深度学习的瓶子检测软件（UI界面+YOLOv5+训练数据集）
摘要:基于深度学习的瓶子检测软件用于自动化瓶子检测与识别,对于各种场景下的塑料瓶.玻璃瓶等进行检测并计数,辅助计算机瓶子生产回收等工序.本文详细介绍深度学习的瓶子检测软件,在介绍算法原理的同时,给出P ...
Pytorch学习记录-torchtext和Pytorch的实例（使用神经网络训练Seq2Seq代码）
Pytorch学习记录-torchtext和Pytorch的实例1 0. PyTorch Seq2Seq项目介绍 1. 使用神经网络训练Seq2Seq 1.1 简介,对论文中公式的解读 1.2 数据预 ...

DeeplabV3+训练数据集流程（学习记录）

一、源码准备

二、训练步骤

DeeplabV3+训练数据集流程（学习记录）相关推荐

最新文章

热门文章