search

mjkvaak / ImageDataAugmentor

Custom image data generator for TF Keras that supports the modern augmentation module albumentations
Other
86 stars 27 forks source link

not work #20

Closed Factral closed 2 years ago

Factral commented 2 years ago

error when declared

dg = ImageDataAugmentor(augment=AUGMENTATIONS, label_augment_mode="mask", preprocess_input=preprocess_input, preprocess_labels=one_hot_encode_masks, seed=SEED )

AttributeError: 'Compose' object has no attribute 'additional_targets'

this is an error that came up when executing this example https://github.com/mjkvaak/ImageDataAugmentor/blob/master/examples/segmentation-with-flow_from_dataframe.ipynb

mjkvaak commented 2 years ago

Thanks for reporting this. I got a different error message that resulted from IDA not supporting newer versions of albumentations. Try if 

pip uninstall ImageDataAugmentor
pip install git+https://github.com/mjkvaak/ImageDataAugmentor

would solve your issue.

Factral commented 2 years ago

thank u!