其中主要通过python爬虫从百度网站中爬取火灾图片及蜡烛图片,而后使用labelImg进行标注,得到xml相关的标注文件,将其转换为Yolo格式,而后使用Yolov5进行网络训练(pytorch框架下,使用GPU),得到模型,编写相应的服务函数,进行检测,其中包括针对视频的及视频流的。
评论