Talk is cheap, Show me the code

0%

U-Net网络训练自己的数据集-踏坑记录

发表于 分类于
本文字数： 301 阅读时长 ≈ 1 分钟

mark一下用U-Net网络训练自己的数据集中遇到的坑

Pytorch U-Net

按照惯例：

GitHub链接：https://github.com/milesial/Pytorch-UNet

论文链接：https://arxiv.org/pdf/1505.04597v1.pdf

参考贴

  1. 代码修改：【记录】使用Pytorch-UNet训练自己的数据集

  2. 准备数据集、训练：U-net复现pytorch版本 以及制作自己的数据集并训练

  3. 多类别训练：Pytorch下实现Unet对自己多类别数据集的语义分割

踏坑过程

  • 先按照参考贴1，训练了细胞的数据集进行了测试，只需要修改一下channel
  • 然后安装labelme准备制作数据集
-------------The End-------------