Skip to content

MLearing/Keras-Unet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

U-net-segmentation

keras / object dection / image segmentation

如果对您有用的话,希望点个赞★!

U-net网络是一个非常强大的分割网络,(其实说它是一个二分类网络更为准确),这个程序中包含了图像数据变换,U-net模型以及结果可视化和图像数据转换保存等功能。我利用U-net实现了对鱼(可拓展到其他物体)边缘检测的结果,结果发现比传统的边缘检测方法的效果要好很多。

U-net网络结构:


程序介绍

  在data/train路径下已经有了image,label两个文件,分别是训练原始图像和ground truth.在以'aug_'开头的文件夹下都有一个test文件,这个只是为了测试,可以直接删除。

程序使用说明:

1:运行data.py,这是为了进行数据增强,原始训练数据太少,所以需要进行仿射变换或者镜像操作等变换方法来生成更多的训练图像和ground truth

2:运行unet.py,这可能会花上一段时间。运行结果会在根目录下出现几个.npy文件。

3:运行test_predict.py,可以将测试集的结果进行可视化。

4:运行data_vision.py,可以将测试集及其结果保存成指定的图片格式,并且保存到指定的路径下。(可以根据自己情况选择)

效果展示:

About

U-Net: Convolutional Networks for Biomedical Image Segmentation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages