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\]提到了yolov4-tiny-pytorch火焰检测框架,其中包括了已经训练好的火焰模型。这个模型是通过训练2000多张火焰数据集而得到的。你可以下载这个框架并直接运行火焰检测。首先,你需要确保你的环境中安装了cuda 10.0,并且可以通过运行"nvcc -V"来检测cuda是否安装完成。然后,你可以下载yolov4-tiny-pytorch火焰检测源码,源码地址可以在引用\[1\]中找到。接下来,你需要在python 3.6环境中安装torch和torchvision库,你可以使用以下命令来安装:pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html。如果你在运行过程中遇到了警告"UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead.",你可以修改model.py文件的代码,在大约192行左右添加以下两句代码:obj_mask=obj_mask.bool()和noobj_mask=noobj_mask.bool()。这样就可以完成yolov4-tiny火灾检测的设置和运行了。 #### 引用[.reference_title] - *1* *2* [yolov4-tiny-pytorch火焰检测,可完全实时检测效果超乎想象!!!](https://blog.csdn.net/liaoqingjian/article/details/120285482)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [Yolov3的Pytorch本实现火焰检测](https://blog.csdn.net/newlw/article/details/124625199)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

萧鼎

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

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

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

打赏作者

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

抵扣说明:

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

余额充值