Keras-yolo3: Syntax error

[lyltencent]

Hi, I have the following error:

$ python train.py
File "train.py", line 130
[*model_body.output, *y_true])
^
SyntaxError: invalid syntax

I am using python2. Is it possible to modify this part to make the code working?

[TianboChen]

you can change the code
from:
model_loss = Lambda(yolo_loss, output_shape=(1,), name='yolo_loss',
arguments={'anchors': anchors, 'num_classes': num_classes, 'ignore_thresh': 0.5})(
[*model_body.output, *y_true])
model = Model([model_body.input, *y_true], model_loss)
to:
lst_tensor = [t for t in model_body.output]
lst_tensor_y_true = [y for y in y_true]
lst_tensor.extend(lst_tensor_y_true)
model_loss = Lambda(yolo_loss, output_shape=(1,), name='yolo_loss',
arguments={'anchors': anchors, 'num_classes': num_classes, 'ignore_thresh': 0.5})(
lst_tensor)
model_tensor = [model_body.input]
model_tensor.extend(lst_tensor_y_true)
model = Model(model_tensor, model_loss)

[lantuxin]

@TianboChen Thank you for your answer. Besides, I modified line 182 of train.py
-yield [image_data, *y_true], np.zeros(batch_size)
+yield [image_data] + y_true, np.zeros(batch_size)