python - 将图像解析为 numpy 数组的问题

Question

我正在尝试将一些图像传递给 numpy 数组以训练深度神经网络。我正在使用 cv2，代码是这样的

def dataloader(filepath, subset):
    # Initiliaze return arrays - input of shape = HYPERPARAMETER
    global size
    if subset=="train":
        size = 129
    elif subset=="test":
        size = 18
    input_data=np.zeros((size,1024,1024,1))
    output_data=np.zeros((size,1024,1024,1))
    # Open file and create loop
    with open(filepath+"annotation_"+subset+".txt", "r") as input_file:
        # Count to pass through the file
        count=0
        for line in input_file:
            line.split()
            input_data[count,:,:,0]=cv2.imread(filepath+str(line[0])+".jpg",cv2.IMREAD_GRAYSCALE)

但这样我的输入都是 nan 我不明白为什么。图像不应该作为我需要的 2d 格式传递。谁能解释为什么会这样？

标签： pythonnumpyopencvimage-processing