dvivarelli
Results
2
comments of
dvivarelli
it works if chage the line in this way: ` img if channels_last else img.transpose(0, 2).transpose(0, 1).numpy() `
If you have CUDA 11.2 like me, you should upgrade dependencies Keras==2.4.3 tensorflow==2.5.0 (tested with python 3.7 and it works) update dipendencies (tested on Colab): ``` astor==0.8.0 cycler==0.10.0 h5py~=3.1.0 Keras==2.4.3...