如何为 DCGAN 代码重塑文件夹中的图像？

如何解决如何为 DCGAN 代码重塑文件夹中的图像？

这些是我文件夹中的 9600 张图片及其属性：enter image description here

我尝试使用下面的代码调整它们的大小，但是这给了我一个 (9600,224,672) 的图像形状。我主要使用此代码是因为它帮助我将图像放入正确的数组中/我之前曾将这段代码用于 kmeans 的事情。 我需要一个更小的 DCGAN 代码的图像形状（更接近 9600,64,128）——原因见下文。

import numpy as np
from numpy import save
import tensorflow as tf
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
from sklearn.metrics import silhouette_score
import cv2
import os,glob,shutil

input_dir = '/Users/User/Documents/MATLAB/attempt6images/'
glob_dir = input_dir + '/*.jpg'
img = [cv2.resize(cv2.imread(file),(224,224)) for file in glob.glob(glob_dir)]
#save('SV_images.npy',images)
paths = [file for file in glob.glob(glob_dir)]
img = np.array(np.float32(img).reshape(len(img),-1)/255)
save('SV_images.npy',img)

我正在尝试通过 DCGAN 代码运行它们，该代码最初是为具有形状 (9600、64、128) 的图像创建的。当我使用图像形状 224,672 时，我的内核死了。所以我假设与原始图像相比，我的图像尺寸可能太大。顺便说一下，我将 25624 部分更改为 256721 以匹配我的图像形状 + 更改了除 256 列之外的所有其他数字）。下面是 64x128 原始代码的一部分。知道如何重塑我的图像大小以便我的内核不会死吗？ enter image description here