win10 yolo3 keras环境搭建

    近期学习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()

 

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值