segmentation_models
segmentation_models copied to clipboard
qubvel
Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
Is there an easy way to select the sizes of the filters in the Unet-resnet? It looks like it starts at 64, and then goes through the U with the...
I am getting a weird loss and IoU score.. Dice loss becomes negative just after 2-3 epochs, and IoU becomes greater than 1 at the same time. Why is this...
I'm trying to apply PSPNet to segment a set of Transmision Xray Images (similar to medical x-ray images but for materials science). I've trained the network with a set of...
def get_training_augmentation(): train_transform = [ A.HorizontalFlip(p=0.5), A.ShiftScaleRotate(scale_limit=0.5, rotate_limit=0, shift_limit=0.1, p=1, border_mode=0), A.PadIfNeeded(min_height=320, min_width=320, always_apply=True, border_mode=0), A.RandomCrop(height=320, width=320, always_apply=True), A.IAAAdditiveGaussianNoise(p=0.2), A.IAAPerspective(p=0.5), A.OneOf( [ A.CLAHE(p=1), A.RandomBrightness(p=1), A.RandomGamma(p=1), ], p=0.9, ), A.OneOf( [...
The default depth of the UNet is 5, which is too deep for images with low resolution. My question is how can I change the depth of the UNet? I...
Hi, Thanks for your great work and it helps me a lot for the segmentation tasks. I find that there are so many implemented classic backbone models in your framework....
Hello I have imported and created unet using below code : - model = Unet(input_shape=(128,128,3),classes=256) but when i run it it gives me error AttributeError: module 'keras.utils' has no attribute...
Hello. I have one question about dice loss. According to these code lines(https://github.com/qubvel/segmentation_models/blob/master/segmentation_models/losses.py#L94:L105), dice loss is calculated by "1 - f_score". So, I understood if f_score is lower, than dice...
I have an issue with when i try to fit my model, i keep getting that says dimensions are not equal Dimension 1 in both shapes must be equal, but...
