from segmentation_models import Unet

model = Unet('resnet34', encoder_weights='imagenet')