Блог

У меня есть несбалансированный набор данных с 04 классами, один из которых является фоновым, а три других класса. Теперь большинство моих изображений содержат фон и еще один класс «x», в то время как два других являются классами меньшинств. Я хотел бы сделать некоторые дополнения только для классов меньшинств. Если я выполню преобразования при создании torchvision.datasets, это увеличит все данные, и мой набор данных все равно будет несбалансированным. Как я должен установить критерии таким образом, чтобы он искал класс меньшинства и выполнял его дополнение? В моем каталоге есть ярлыки, изображения и основные истины. Поскольку я использую инструмент создания альбомов для обрезки и изменения размера, я подумываю о том, чтобы использовать его для переворачивания и преобразования изображений для класса меньшинства, но я не понимаю, как искать изображения с классом меньшинства и применять преобразование только к этим изображениям.