Unable to allocate 21.4 GiB for an array with shape (14630, 256, 256, 3) and data type float64
借鉴博客 https://blog.csdn.net/qq_37591637/article/details/103349269?depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-1&utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-1