Deep Learning - Transfer learning

[lao8n]

Hello,

I tried to follow the Transfer learning example in pyspark on page 534-535. However, when I try to do p_model = p.fit(train_df) I get a 'Number of source Raster bands and source color space components do not match'. I tried googling what to do, but i wasn't sure what to do? The trace includes references to java.awt.image.ColorConvertOp.filter, com.sun.imageio.plugins.jpeg.JPEGImageReader.... and javax.imageio.ImageIO.read. Note that I used Spark ml's ImageSchema read function because the sparkdl readImages function seems to be deprecated.

Thanks in advance,
Nick