Question

我正在尝试为CCN问题扩充数据。我有一个带有3个图像（中央，左侧，右侧）的csv，并且转向角对于所有三个图像都相同。（我使用了Udacity自动驾驶汽车模拟器）。我想增加数据，因此尝试对图像进行扩充。我需要创建一个具有如下形状的元组：

image_path = data[["center", "left", "right"]].values

但是在此之前，我必须扩充我的数据。那是我的代码：

steerings = data["steering"].values
steerings = list(steerings)
center = data["center"].values
left = data["left"].values
right = data["right"].values
center = list(center)
new_center = []
left = list(left)
new_left = []
right = list(right)
new_right = []
new_steerings = []
for index in range(len(center)):
    new_center_img = augment_image(center[index], steerings[index])[0]
    new_center.append(new_center_img)
    new_left_img = augment_image(left[index], steerings[index])[0]
    new_left.append(new_left_img)
    new_right_img = augment_image(right[index], steerings[index])[0]
    new_right.append(new_right_img)
    new_steerings.append(steerings[index])
center.extend(new_center)
left.extend(new_left)
right.extend(new_right)
steerings.extend(new_steerings)

image_path = tuple(center), tuple(left), tuple(right)
steerings = tuple(steerings)


return train_test_split(image_path, steerings, test_size=0.2, random_state=1)

但是：

image_path = tuple(center), tuple(left), tuple(right)

不同于：

image_path = data[["center", "left", "right"]].values

返回图像路径时如何获得相同的东西，但由于扩增而出现双重图像？

Answer 1

如果我理解正确，我认为您只需要这样做：

image_path = np.stack([center, left, right], axis=1)

np是NumPy。

如何创建具有另一个形状的元组

1 个答案: