python (12.9k questions)

javascript (9.2k questions)

reactjs (4.7k questions)

java (4.2k questions)

java (4.2k questions)

c# (3.5k questions)

c# (3.5k questions)

html (3.3k questions)

Questions - albumentations

How resize dataset label in albumentations label to work with tensorflow image_dataset_from_directory function?

I am running the following code: [https://pastebin.com/LK8tKZtN] The error obtained is following: File "C:\Users\Admin\PycharmProjects\BugsClassfications\main2.py", line 45, in set_shapes ...
test-img

Ion Caciula

python

tensorflow

keras

image-augmentation

albumentations

Votes: 0

Answers: 1

Latest Answer

If you change the shape of your labels, it should work: def set_shapes(img, label, img_shape=(128,128,3)): img.set_shape(img_shape) label.set_shape([1,]) return img, label But you should ask yo...
test-img

AloneTogether

Posts

Questions

Blogs

Jobs