Skip to content

一个计算机视觉深度神经网络效果可视化GUI程序,其支持ONNX格式的神经网络文件。

License

Notifications You must be signed in to change notification settings

wanderyum/EffectViewer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Effect Viewer

Effect Viewer是一个计算机视觉深度神经网络效果可视化GUI程序,其支持ONNX格式的神经网络文件。

Effect Viewer支持以下输入:

  • 本地图片(png/jpg格式)
  • 摄像头输入

Effect Viewer的使用:编译好之后点点点就行了。

效果演示

采用Yolo v4网络进行目标检测。

编译过程

安装依赖

编译

  1. 设置CMakeLists.txt中各库的路径
    打开CMakeLists.txt,修改“# 设置路径”这一部分内容,其中:
    • Qt5_DIR是Qt5所在目录
    • OpenCV_DIR是Opencv所在目录
    • ORT_INCLUDE_DIR是ONNX Runtime的头文件目录
    • ORT_LIB_DIR是ONNX Runtime库文件所在目录
  2. 在根目录中新建build文件夹并进入,命令行中输入:
    cmake ..
    cmake --build .
  3. onnxruntime.dll和ONNX格式的神经网络(比如yolov4.onnx)拷贝到build/Debug/文件夹中。
  4. build/Debug/文件夹中运行EffectViewerEffectViewer.exe

To-do list:

  • 数据源中增加“视频”这一类型
  • 增加“类别生成器”模块,使其能够通过读取配置文件自动生成类文件
  • 任务类型(MissionType)添加对图片分类、风格迁移等类型任务的支持
  • 语言国际化
  • 增加更多模型

About

一个计算机视觉深度神经网络效果可视化GUI程序,其支持ONNX格式的神经网络文件。

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published