如何通过增强来平衡文件夹中的图像,使该文件夹中的图像数量等于其他文件夹中的图像数量?

How to balance images in folder by doing augmentation such that number of images in this folder are equal to number of images in other folder?

我有 5 个名为 class_i 的文件夹,每个文件夹都有 i class 图片。图像为.jpg 格式。 如何通过增强来平衡每个文件夹中的图像,使该文件夹中的图像数量等于图像数量最多的文件夹中的图像数量? 另外,能否请您帮忙绘制一条曲线,显示平衡前后每个文件夹中的图像数量?

刚刚扩展了 ,算法完全符合您在本题中的要求。算法位于我答案的最后。

它接受两个文件夹 - 一个用于输入,一个用于输出。每个文件夹包含以整数标签命名的子文件夹,每个子文件夹包含该标签的图像。

当代码为 运行 时,它首先进行增强,然后绘制输入文件夹(未修改)的图像计数,并在 window 关闭后绘制第二个 window 和增强计数(输出)图像。

在输出文件夹中,每个图像都以源图像名称加上表示用于生成输出 PNG 图像的增强方法的后缀命名。每个图像可以有多个增强过滤器,每个过滤器产生一个 PNG 输出。