需要有关在 segmentation_models API 中使用预训练权重的指导
need guidance on using pre-trained weights in segmentation_models API
我想使用使用 segmentation_models API for the Cityscapes 数据集的预训练 Unet 模型,但我需要相同的预训练权重。我在哪里可以找到在 Cityscapes 数据集上训练的 Unet 模型的预训练权重?
请指导我!!!
Google 的 benchmark so i assume it is not adapted for this dataset (too slow and not enough performant probably). However, I advise you to start with DeepLabv3+ 中没有 UNet,它并不复杂,更适合这个数据集。
您可以使用这个 repository where it is implemented, well documented and useable with pretrained weights from cityscape dataset (and also PascalVOC 数据集)。
我想使用使用 segmentation_models API for the Cityscapes 数据集的预训练 Unet 模型,但我需要相同的预训练权重。我在哪里可以找到在 Cityscapes 数据集上训练的 Unet 模型的预训练权重?
请指导我!!!
Google 的 benchmark so i assume it is not adapted for this dataset (too slow and not enough performant probably). However, I advise you to start with DeepLabv3+ 中没有 UNet,它并不复杂,更适合这个数据集。
您可以使用这个 repository where it is implemented, well documented and useable with pretrained weights from cityscape dataset (and also PascalVOC 数据集)。