Skip to content

liandun110/fire_detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

说明

  • 算法:YOLOv5

使用pytorch模型进行fire detection

python detect.py --weights model.pt --device=cpu --source=img.png

将pytorch模型转换成onnx模型

python export.py --weights model.pt --include onnx --opset 16

使用onnx模型进行fire detection

python detect.py --weights model.onnx --device=cpu --source=img.png

使用onnx模型在一个文件里独立进行fire detection

python detect_onnx.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages