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