keras训练完以后怎么预测_Tips: keras使用多gpu训练（以yolov3为例）

参考：

qqwweee/keras-yolo3​github.com

在train.py中

1.首先开头加入：from keras.utils import multi_gpu_model

2.找到create_model函数观察，在以下两句之后

 model_body = yolo_body(image_input, num_anchors//3, num_classes)
 print('Create YOLOv3 model with {} anchors and {} classes.'.format(num_anchors, num_classes))

添加：

model_body = multi_gpu_model(model_body,gpus=2)

gpus=2表示使用两块gpu，/gpu:0 /gpu:1

注意，如果multi_gpu_model加错地方，就会报错如Can't concatenate scalars (use tf.stack instead) for 'yolo_loss_1/concat' (op: 'ConcatV2') with input shapes: [], [], []

此外，如果出现如下错误

python3.5/site-packages/tensorflow/python/client/session.py", line 712, in __del__
python3.5/site-packages/tensorflow/python/framework/c_api_util.py", line 31, in __init__

可以在代码开头加上

from keras import backend as K
K.clear_session()