Требуется руководство по использованию предварительно обученных весов в API segmentation_models
Я хочу использовать предварительно обученную модель Unet с использованием segmentation_models API для набора данных Cityscapes, но для этого мне нужны предварительно обученные веса. Где я могу найти предварительно обученные веса для модели Unet, обученной на наборе данных Cityscapes?
Пожалуйста, подскажите мне в этом!!!
Ответы 1
UNet отсутствует в тесте , поэтому я предполагаю, что он не адаптирован для этого набора данных (возможно, слишком медленный и недостаточно производительный). Однако я советую вам начать с DeepLabv3+ от Google, которая не так сложна и более адаптирована для этого набора данных.
Вы можете использовать этот репозиторий , где он реализован, хорошо задокументирован и пригоден для использования с предварительно обученными весами из набора данных городского пейзажа (а также набора данных PascalVOC).