当前位置:   article > 正文

『paddle』paddleseg 学习笔记:代码结构

『paddle』paddleseg 学习笔记:代码结构

目录


在尝试完成使用配置文件进行训练之后,肯定有小伙伴想基于 PaddleSeg 进行更深入的开发,在这里大概介绍一下 PaddleSeg 代码结构。

PaddleSeg
     ├──  configs 	# 配置文件文件夹
     ├──  paddleseg # 训练部署的核心代码
        ├── core  
        	├── infer.py
            └── predict.py
            └── train.py
            └── val.py
        ├── cvlibs 	# Config 类定义在该文件夹中。它保存了数据集、模型配置、主干网络、损失函数等所有的超参数。
            ├── callbacks.py
            └── ...
        ├── datasets 	# PaddleSeg 支持的数据格式,包括 ade、citycapes 等多种格式
            ├── ade.py
            ├── citycapes.py
            └── ...
        ├── models 		# 该文件夹下包含了 PaddleSeg 组网的各个部分
            ├── backbone 	# paddleseg 的使用的主干网络
            ├── hrnet.py
            ├── resnet_vd.py
            └── ...
            ├── layers 	# 一些组件,例如 attention 机制
            ├── activation.py
            ├── attention.py
            └── ...
            ├── losses 	# 该文件夹下包含了PaddleSeg所用到的损失函数
            ├── dice_loss.py
            ├── lovasz_loss.py
            └── ...
            ├── ann.py 		# 该文件表示的是PaddleSeg所支持的算法模型,这里表示ann算法。
            ├── deeplab.py 	# 该文件表示的是PaddleSeg所支持的算法模型,这里表示Deeplab算法。
            ├── unet.py 	# 该文件表示的是PaddleSeg所支持的算法模型,这里表示unet算法。
            └── ...
        ├── transforms 		# 进行数据预处理的操作,包括各种数据增强策略
            ├── functional.py
            └── transforms.py
        └── utils
            ├── config_check.py
            ├── visualize.py
            └── ...
     ├──  train.py  	# 训练入口文件,该文件里描述了参数的解析,训练的启动方法,以及为训练准备的资源等。
     ├──  predict.py 	# 预测文件
     └── ...
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/510380
推荐阅读
相关标签
  

闽ICP备14008679号