Skip to content

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

@GuidoA

Description

@GuidoA

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions