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