本文已参与「新人创作礼」活动,一起开启掘金创作之路。
该项目的数据集使用的鱼,对鱼进行分割。由于数据集的数量太少,所以先进行数据增强,进行仿射变换或者镜像操作等变换方法来生成更多的训练图像。该项目来源于:github.com/shuyucool/U…
下面是该文件的文件结构以及README文件的内容
DEBUG:
问题出现在生成训练图像的阶段,运行data.py文件时,图像能够顺利增强,但是不能够将aug_merge中的图像拆分到aug_label和aug_train中。
运行data.py后aug_merge中正确生成了文件夹,并且文件夹中的生成了增强后的图像,多少张图像就生成多少个文件夹。
但是aug_label和aug_train两个文件夹中只生成了相应的文件夹,文件夹中并没有图像。
有此推断问题应该出现在拆分的代码处,以下为拆分方法的代码,在网上找了几个U-net的代码,都是这样写的,但是我跑不出来结果:
经过排错发现是在上图的最后一步拆分出现了问题,数据没有保存成功:
加了两句输出看下保存路径,可以发现保存路径出现了问题。
针对上述问题进行修改,去掉str(i),之后便能正常拆分数据并保存。