我有一个不平衡的数据集，其中包含 04 个类，其中一个是背景类，另外三个是其他类。现在，我的大部分图像都包含背景和另一个类“x”，而另外两个是少数类。我想只对少数类做一些扩充。如果我在创建 torchvision.datasets 时执行转换，它将增加所有数据，并且我的数据集仍然不平衡。我应该如何设置标准，以便它寻找少数类并对其进行扩充？我的目录中有标签、图像和基本事实。当我使用标注工具进行裁剪和调整大小时，我正在考虑使用它来翻转和转换少数类的图像，但我不明白如何查找具有少数类的图像并将转换仅应用于这些图像。