Skip to content

用Pytorch复现常用语义分割网络,备用

Notifications You must be signed in to change notification settings

ShaoYULi12/Pytorch-SegToolbox

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

61 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch-SegToolbox

用Pytorch复现常用语义分割网络,备用

目前实现DeepLabV3+、UNet

目录结构

Pytorch-SegToolbox
  |--backbones 存放主干网络ResNet、AlignedXception等
  |--datasets 存放数据集、观察数据工具等
      |--data_list 存放数据集的csv文件
  |--models 存放语义分割网络 DeepLabV3+(Xception|ResNet101)、UNet(ResNet)、FCN8s等
  |--res 存放资源 权重文件、图像、日志等
  |--utils 存放工具
      |--augment数据增强
      |--lossfn损失函数
      |--tools日志、计时、评价标准metrics等工具
  |--test.py 用来测试
  |--train.py 用来训练

权重文件&测试文件

由于训练资源(租用GPU)太烧钱,只训练了DeepLabV3+(Xception)和FCN8s。 权重文件,以及以DeepLabV3+(Xception)的测试输出的图片可以在此处下载:
链接:https://pan.baidu.com/s/1U0lx83uKexpfYPaTen3vog
提取码:iyqd
res下的preds只展示了几张效果不错的测试输出,效果不好的,我就不贴了。哈哈……

About

用Pytorch复现常用语义分割网络,备用

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%