ZZUTK / Face-Aging-CAAE

Age Progression/Regression by Conditional Adversarial Autoencoder
636 stars 233 forks source link

[In-the-wild Faces Dataset] Removing `alpha` channel in `load_image()` #16

Open RooseveltAdvisors opened 7 years ago

RooseveltAdvisors commented 7 years ago

Training failing on In-the-wild Faces dataset which you can find the download link in the Datasets section of this site (https://susanqq.github.io/UTKFace/)

The error message I saw was as follows:

could not broadcast input array from shape (128,128,3) into shape (128,128)

This error could potentially result from load_image returning a M N 4 matrix if there is a RGBA image in the batch.