需要有关在 segmentation_models API 中使用预训练权重的指导 [关闭]

Posted 2023-03-27

【中文标题】需要有关在 segmentation_models API 中使用预训练权重的指导 [关闭]

【英文标题】：need guidance on using pre-trained weights in segmentation_models API [closed]

【发布时间】：2021-03-22 23:15:12

【问题描述】：

我想为 Cityscapes 数据集使用使用 segmentation_models API 的预训练 Unet 模型，但我需要相同的预训练权重。在哪里可以找到在 Cityscapes 数据集上训练的 Unet 模型的预训练权重？

请指导我！！！

【参考方案1】：

benchmark 中没有 UNet，所以我认为它不适合此数据集（可能太慢且性能不够）。不过，我建议您从 Google 的 DeepLabv3+ 开始，它并不复杂，更适合此数据集。

您可以使用这个 repository 来实现它，它有据可查，并且可与来自城市景观数据集（以及 PascalVOC 数据集）的预训练权重一起使用。