运行U-net网络代码时出现的错误

187 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

该项目的数据集使用的鱼,对鱼进行分割。由于数据集的数量太少,所以先进行数据增强,进行仿射变换或者镜像操作等变换方法来生成更多的训练图像。该项目来源于:github.com/shuyucool/U…

在这里插入图片描述 下面是该文件的文件结构以及README文件的内容 在这里插入图片描述

DEBUG:

问题出现在生成训练图像的阶段,运行data.py文件时,图像能够顺利增强,但是不能够将aug_merge中的图像拆分到aug_labelaug_train中。

运行data.pyaug_merge中正确生成了文件夹,并且文件夹中的生成了增强后的图像,多少张图像就生成多少个文件夹。 但是aug_labelaug_train两个文件夹中只生成了相应的文件夹,文件夹中并没有图像。 aug_merge文件夹aug_merge文件夹中的数据aug_label文件夹aug_train文件夹 有此推断问题应该出现在拆分的代码处,以下为拆分方法的代码,在网上找了几个U-net的代码,都是这样写的,但是我跑不出来结果: 在这里插入图片描述 经过排错发现是在上图的最后一步拆分出现了问题,数据没有保存成功: 加了两句输出看下保存路径,可以发现保存路径出现了问题。 在这里插入图片描述在这里插入图片描述

针对上述问题进行修改,去掉str(i),之后便能正常拆分数据并保存。 在这里插入图片描述在这里插入图片描述在这里插入图片描述