Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training on custom dataset, I get AttributeError: 'Tensor' object has no attribute 'numpy' #136

Open
GuidoA opened this issue Feb 13, 2020 · 1 comment

Comments

@GuidoA
Copy link

GuidoA commented Feb 13, 2020

Hello, and thanks for making a Keras version of DeepLab!
I've been trying to train your model on my custom dataset. All my images are 1088x2048x3, and I only have two classes. When I try to train the model on my dataset, I incur in these warnings:

Guido_warnings

which are then followed by this Traceback:

Guido_error

These are the loss, metric, and model.compile I'm using:

Guido_losses

and these are the generators and model.fit_generator I'm using:

Guido_code

Do you have any idea what may be causing this? Would greatly appreciate any help!
Thanks.

@aliFrancis
Copy link

I have the same issue using fit_generator with a custom loader. Have you found a solution?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants