Я довольно новичок в PyTorch, но я смог приступить к обучению и тестированию предварительно обученных моделей и моих собственных наборов данных только обычных классификаций изображений.Однако сейчас я хочу опробовать семантическую сегментацию.У меня есть набор данных с входными изображениями, которые выглядят следующим образом:
и метки истинности, которые выглядят следующим образом (сгенерированные с использованием сегментации в MATLAB):
Как обучить модели семантической сегментации в PyTorch, используя мой собственный набор данных с изображениями / метками, такими как эти?Любая помощь будет оценена!