Как использовать Torchvision Mixup для решения проблемы с несколькими метками?Python

Программы на Python
Anonymous
Как использовать Torchvision Mixup для решения проблемы с несколькими метками?

Сообщение Anonymous »

Я хочу использовать MixUp в качестве дополнения к данным для задачи классификации изображений с несколькими метками, но кажется, что он поддерживает только метки с формой (batch_size,). Как мне изменить это, чтобы разрешить использование таких меток, как [0.0, 1.0, 0.0, 1.0]
Я пробовал вызвать argmax для моего вектора меток [0.0, 1.0] , 0.0, 1.0], что дает мне один индекс. Это заставляет функцию работать, но она смешивает только один класс.

Подробнее здесь: https://stackoverflow.com/questions/784 ... el-problem

Вернуться в «Python»