yolov4-tiny-pytorch版的火焰检测,可完全实时检测,效果超乎你想象!!!

yolov4-tiny-pytorch版的火焰检测,框架中包括已经训练好的火焰模型,该火焰模型是通过训练2000多张火焰数据集训练而成,下载后可直接运行检测
1.使用环境cuda 10.0,环境装好后使用nvcc -V检测,出现以下说明cuda安装完成
在这里插入图片描述
2.下载框架,yolov4-tiny-pytorch版的火焰检测源码,框架中包括已经训练好的火焰模型,
源码地址在最下方!

3.使用python 3.6环境,下载重要的torch和torchvision库,使用命令:

pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

4.下载其他的库:

pip install numpy==1.16.1
pip install opencv-python
pip install tqdm
pip install Pillow

5.环境配置好后,找到video.py
在这里插入图片描述

6.运行video.py文件,展示效果,帧率基本上在30fps,可实时检测。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

在这里插入图片描述

7.想要训练的依次运行
voc2yolo4.py、voc_annotation.py和train.py文件即可开始训练
在这里插入图片描述

在这里插入图片描述

9.最后,源码地址:https://download.csdn.net/download/liaoqingjian/22761642?spm=1001.2014.3001.5501
该数据源码包括数据集和火焰模型,需要9.9元,训练标注不易,希望理解。

  • 1
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 10
    评论
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

萧鼎

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值