遇到这个问题,首先检查一下cv2的imread的读取路径对不对,如果正确的话,就可以按照我下面的方式在原代码修改一下
def data_clip_generator():
while True:
data_clip = []
frame_id = rng.randint(0, length-1)
#######inputs
data_clip.append(np_load_frame(data_info_list[1]['frame'][frame_id], 384, 512))
data_clip.append(np_load_frame(data_info_list[2]['frame'][frame_id], 384, 512))
data_clip.append(np_load_frame(data_info_list[0]['frame'][frame_id], 384, 512))
data_clip = np.concatenate(data_clip, axis=2)
yield data_clip
修改之后的:
def np_load_frame(filename, resize_height, resize_width):
# 读取图像
image_decoded = cv2.imread(filename)
# 检查图像是否正确加载
if image_decoded is None:
raise ValueError("Unable to load image at path: " + filename)
# 如果指定了resize_height,则调整图像大小
if resize_height is not None:
image_resized = cv2.resize(image_decoded, (resize_width, resize_height))
else:
image_resized = image_decoded
# 将图像转换为float32类型,并归一化
image_resized = image_resized.astype(dtype=np.float32)
image_resized = (image_resized / 127.5) - 1.0
return image_resized
然后运行一下,看看错误还有没有