Unet_erythema_detection
A basic neural network for image segmentation.
前期准备
1.在logs文件夹中下载h5权重文件,百度网盘链接在logs文件夹中
2.将所有原图 放置在“/dataset_1/JPEGImages/”文件夹中,将json转换的所有图片 放置在“/dataset_1/SegmentationClass/”文件夹中
搭载Colab使用
- 注册谷歌账号,进入谷歌云端硬盘
- 上传文件夹,进入后新建一个 ==Google Colaboratory== 3.设置GPU:菜单栏——代码执行程序——更改运行时类型——选择GPU、 4.连接谷歌云端文件夹 5.运行训练程序 6.预测红斑区域 ==若有报错“找不到改文件”,需要复制.py文件完整路径再运行==
注意事项
train.py里数据集的路径根据每个人电脑上colab中dataset_1文件夹的路径不同进行修改