albumentations-team / albumentations

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
https://albumentations.ai
MIT License
14.09k stars 1.64k forks source link

Lambda #732

Open ruchika61 opened 3 years ago

ruchika61 commented 3 years ago

hi getting this error , kindly help def get_preprocessing(preprocessing_fn): """Construct preprocessing transform

Args:
    preprocessing_fn (callbale): data normalization function 
        (can be specific for each pretrained neural network)
Return:
    transform: albumentations.Compose

"""

_transform = [A.Lambda(image=preprocessing_fn)]
return A.Compose(_transform)

############################################################################## File "", line 1, in runfile('E:/MCSS/segmentation_models-master/load_dataset.py', wdir='E:/MCSS/segmentation_models-master')

File "C:\Users\dell\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 704, in runfile execfile(filename, namespace)

File "C:\Users\dell\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 108, in execfile exec(compile(f.read(), filename, 'exec'), namespace)

File "E:/MCSS/segmentation_models-master/load_dataset.py", line 292, in preprocessing=get_preprocessing(preprocess_input),

File "E:/MCSS/segmentation_models-master/load_dataset.py", line 241, in get_preprocessing _transform = [A.Lambda(image=preprocessing_fn)]

AttributeError: module 'albumentations' has no attribute 'Lambda'

Dipet commented 3 years ago

Try to update albumentations to latest version.

ruchika61 commented 3 years ago

Thanks, it worked