Views
No views yet
imagenet pre-trained weights for the vgg11_bn model used as
encoder in the segmentation-models-pytorch library.pip install segmentation-models-pytorch1import segmentation_models_pytorch as smp
2
3model = smp.Unet("vgg11_bn", encoder_weights="imagenet")