Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
exp		exp
log		log
nets		nets
utils		utils
LICENSE		LICENSE
README.md		README.md
export.ipynb		export.ipynb
train.py		train.py

Repository files navigation

pytorch-deeplabv3-training

用你的数据集训练DeepLabV3

请根据https://github.com/bubbliiiing/deeplabv3-plus-pytorch/ 制作数据集并放入VOCdevkit文件夹

根据数据集修改train.py里面的设置

运行python train.py

导出ONNX（Labview ONNX工具包推理使用的格式）请查看export.ipynb

TODO

增加MSCOCO数据集支持
增加推理代码
增加Paddle Inference导出

感谢 bubbliiiing | msminhas93