Как я могу исправить проблему, чтобы изменить процесс в изображении, полученном из x_train в Python?

Question

Я нашел набор данных от Kaggle Вот ссылка: https://www.kaggle.com/quangqiyana/human-gender-identity

Я хочу реализовать алгоритм CNN в наборе данных.

Я написал несколько кодов, чтобы получить X_train и Y_train

train = pd.read_csv("files/gender.csv")
train.shape -> (230, 67502)
train.drop('Unnamed: 0', axis=1, inplace=True)
Y_train = train["Label"]
X_train = train.drop(labels = ["Label"],axis = 1) 

Затем я хочу показать некоторые изображения с помощью ilo c process

img = X_train.iloc[0].to_numpy()
img = np.pad(img, (0, (67600-img.shape[0])), 'constant').reshape((260, 260))
plt.imshow(img)
plt.title(train.iloc[0,0])
plt.axis("off")
plt.show()

Поскольку любое число не является квадратом 67502, я могу использовать pad. Но изображение не могло отображаться с разрешением.

Вот скриншот.

введите описание изображения здесь

Как исправить проблему с изменением формы?

Answer 1

Этот набор данных, вероятно, не предназначен для использования с CNN, потому что данные, закодированные в столбцы, не имеют пространственного отношения друг к другу, как в изображениях. Учитывая, что этот набор данных был загружен 1 (один) раз, вероятно, вами, и никто не создавал никаких записных книжек и не считал его стоящим обсуждения, я бы рекомендовал перейти к другому набору данных, над которым работают другие люди, чтобы вы могли задайте вопросы там (на Kaggle) и получите помощь.