如何在4D numpy阵列上调整图像大小

2024-09-29 21:28:24 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个4D numpy数组,它存储了50张100*100的图像。所以我想用它们来训练3d Resnet CNN。如何调整Resnet输入的图像大小?你知道吗

原始形状为:

data.shape

(50, 128, 100, 100)

目标形状为:

(50, 128, 96, 96)

输入为:

model = Resnet3DBuilder.build_resnet_18((128, 96, 96, 1), 20)

slice = 128
img size = 96*96*1
classes = 20

Tags: 图像buildnumpy目标imgdatamodelslice

热门问题