近期学习yolo3的目标检测算法,参考了https://github.com/bubbliiiing/yolo3-keras此代码,但是环境一直还是之前的python3.7的环境,作者的环境为py3.6,tf1.13,keras2.1.5,因为要使用tensorflow-gpu,一直也没搭建成功,后来把keras改为2.2.4成功了,过程如下:
1.conda创建py36的环境,可参考https://blog.csdn.net/mmmbox/article/details/103995748将python改成3.6即可
conda create -n py36 python=3.6
2.安装keras和tensorflow执行
conda install cudnn=7.3.1 tensorflow-gpu=1.13 keras=2.2.4
注如果使用tensorfflow cpu的版本,可直接执行
conda install keras=2.2.4 cudnn=7.3.1
将keras改成2.1.5也没问题
3.运行pycharm,配置参见https://blog.csdn.net/mmmbox/article/details/104024277
测试,正常的话将显示yolo3的网络结构
from nets.yolo3 import yolo_body
from keras.layers import Input
inputs = Input([416,416,3])
model = yolo_body(inputs,3,20)
model.summary()