图像分割
该项目在实现了用于语义分割的神经网络。
项目概况
该项目的主文件是convolutional_autoencoder.py ，其中包含用于数据集处理的代码（数据集类），模型定义（模型类）以及训练代码。
为了抽象模型中的图层，我们创建了layer.py类接口。 此类当前具有两个实现： conv2d.py和max_pool_2d.py 。
要推断训练后的模型，请查看infer.py文件。
最后，有几个文件夹：
数据*包含预处理的数据集（请注意，当前模型实现应至少与128x128图像一起使用。）
imgaug包含用于数据扩充的代码（ ）
noteboks包含一些有趣的图像分割
评论0