隨著社會經濟的高速發展，工業、企業園區，住宅日益增多，存在一些消防安全隱患，導致火災事故屢屢發生，
針對住宅，工業園區，森林，加油站等室內外場景，深入了解，并通過目標檢測演算法，在影像中自動標注疑似煙霧、火焰，實作機器智能值守，具有重大意義，
運行環境是pytorch1.81 gpu，可以在免費云gpu上直接運行，一共三個壓縮包
yolov5.zip.mp4(原始碼和運行配置word)
fire_smoke_labels.zip.mp4(資料集的標簽檔案)
small_dataset.zip.mp4(資料集圖片煙霧和火焰分別標注2類，一共2000張)
因為網路問題無法上傳zip檔案
需要把.mp4后綴去掉改為zip檔案才能打開，
云注冊地址
https://gpushare.com/auth/register?user=18*****1785&fromId=c4100216220&source=link
具體云gpu環境配置看word檔案，
1環境安裝
2安裝wandb及訓練，安裝wandb具體細節可以查csdn上的教程，
3 訓練后的map@0.5大概55多
加大訓練批次還能繼續提升，
4推斷視頻
5運行結果
三個下載鏈接
原始碼yolov5.zip.mp4
資料集標簽fire_smoke_labels.zip.mp4
資料集圖片small_dataset.zip.mp4
測驗效果視頻
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/276159.html
標籤：其他