train_data=[]
with open('/home/easofy/Downloads/structured_images/triplets.csv', 'r') as fp:
    next(fp)
    reader = csv.reader(fp)
    for line in reader:
        a=np.array(cv2.imread(line[0],cv2.IMREAD_UNCHANGED))

train_data.append([a, b, c])

    if len(train_data)==6:
        break

[[155, 189, 218],