热门标签 | HotTags
当前位置:  开发笔记 > 编程语言 > 正文

基于Keras和TensorFlow在Windows上实现YOLOv3目标检测

本文详细介绍了如何在Windows环境下配置GPU支持,并使用Keras和TensorFlow实现YOLOv3模型进行图像目标检测。对于环境搭建的具体步骤,可参考外部链接提供的指南。

为了在Windows环境下成功运行YOLOv3模型,需要正确配置GPU支持。具体的配置方法可以在这里找到详细的指南。

本项目使用了qqwweee在GitHub上开源的keras-yolo3代码库,项目地址为:https://github.com/qqwweee/keras-yolo3

我的实际配置情况如下:
  • 操作系统:Windows 10
  • TensorBoard 版本:1.8.0
  • TensorFlow-GPU 版本:1.8.0
  • 安装命令:pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ tensorflow-gpu==1.8.0
  • Keras 版本:2.2.4
  • 安装命令:pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ keras==2.2.4
  • Python 版本:3.6.8
  • CUDA 版本:9.0
  • CUDA 下载地址:NVIDIA CUDA 9.0

通过上述配置,能够确保YOLOv3模型在Windows 10系统上的顺利运行,同时利用GPU加速模型训练过程,提高训练效率。


推荐阅读
author-avatar
重庆制造漫画社
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有