耶律大石的博客

YOLOV4：You Only Look Once目标检测模型在Keras当中的实现 目录 实现的内容 Achievement 所需环境 Environment 注意事项 Attention 小技巧的设置 TricksSet 文件下载 Download 预测步骤 How2predict 训练步骤 How2train 参考资料 Reference 实现的内容 主干特征提取网络：DarkNet53 => CSPDarkNet53 特征金字塔：SPP，PAN 训练用到的小技巧：Mosaic数据

一、 问题描述 最近在跑yolo v3的多目标跟踪，环境：win10+tensorflow+ python +keras 出现以下错误： raise TypeError(‘Keyword argument not understood:’, kwarg) 解决Keras加载模型TypeError: (‘Keyword argument not understood:’, u’return_state’) 二、原因分析 由于使用了安装了tensorflow，tensorflow自带了keras，所以在自己安装

keras 运行yolo v3 出现Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED问题， 在代码最开始加入以下代码： import tensorflow as tf from keras import backend as K config = tf.ConfigProto() config.gpu_options.allow_growth=True sess = tf.Session(config=config) K.set_session

# import matplotlib.pyplot as plt # # history = model.fit(x, y, validation_split=0.25, epochs=50, batch_size=16, verbose=1) # # # 绘制训练 & 验证的准确率值 # plt.plot(history.history['acc']) # plt.plot(histo...