Я пытаюсь обучить головку FCOS и отдельную головку классификации, которые используют общую основу. Поскольку я имею дело с наборами данных маммографии, которые содержат большую часть отрицательных образцов (изображений без объектов), простого обучения с положительными образцами может быть недостаточно. Поэтому я решил включить в свой обучающий набор отрицательные образцы. Однако когда я передаю ограничивающую рамку отрицательного образца как пустой torch.tensor, FCOS выдает ошибку.
Код: Выделить всё
Expected target boxes to be a tensor of shape [N, 4], got torch.Size([0])
Мне все еще нужно выполнить классификацию этих изображений с помощью второй головки, и я боюсь, что классификационная головка будет необъективной, если я удалю эти отрицательные образцы.
Предлагаете ли вы метод включения отрицательных образцов на FCOS или мне следует их удалить?
Подробнее здесь:
https://stackoverflow.com/questions/785 ... ve-samples