Unet-pytorch 西瓜叶片图像分割模型

Unet-pytorch

—————————
|GITHUB 项目地址 |
—————————
我的输入数据尺寸：（512*512）

自然环境以及白背景下单片西瓜叶片分割模型以及利用MIOU（平均交并比）算法进行评估、分割结果展示
(由于数据集未全部上传，运行本项目代码的train.py或者accuracy.py文件皆有可能会报错，建议自己替换数据集并修改代码!)

src目录

存放Unet模型代码及训练、预测代码，由于训练数据为我们学校私有，不便全部上传，仅上传18张原图片与标注图以便测试

---

Unet_model.py

    Unet模型构建代码保存在此文件内
    实例化Unet类时传入两个参数
        1：输入图片的通道数
        2：输出图片的通道数

---

loaddata.py

    利用 torch.utils.datar 中的 Dataset,DataLoade 来读取数据，构建数据导入器  
    由于数据集不可外传，DealDat