WIN10+CUDA10.1环境下Keras-YoloV3训练教程环境配置准备过程数据集标注过程数据集准备过程:一二训练过程测试过程最后
环境配置
CUDA和Keras的安装可以参考我的上一篇博客这是链接。
需要注意的是,Tensorflow有GPU和CPU两个版本。
如果我们同时安装了CPU和GPU版本,此时安装Keras,会默认安装CPU的版本的Keras,无法使用GPU进行加速训练。
解决方法就是先卸载Tensorflow,只用pip安装GPU版本,然后再pip安装Keras即可。
准备过