Unet 细胞分割

pytorch训练网络需要注意的细节（以及模板）
笔记：https://www.zybuluo.com/CQUyh/note/2234948 标签（空格分隔）： Pytorch 以细胞分割图为例（Unet和Convnext）
1.model的搭建
这个不多说，是独立的模块，只需要注意输出输出是什么，输入通道和输出通道；一个小技巧，使用monai的框架，十分方便，但是注意load_weights时读取方式。
2.整个代码架构
model、dataset作为存放模型的文件，其中每一个epcoh（训练和测试）单独写在utils，这是工具文件夹。而整体的代码写在跟目录的train.py

train.py:
①路径以及文件读取
②transform和dataset、dataloader
③定义模型，损失函数，优化器
④定义for训练，epcoh的for循环，最大层
main函数里面设置参数

Unet 细胞分割最先出现在Python成神之路。