美文网首页
yolo v3 keras版本增加图形界面UI

yolo v3 keras版本增加图形界面UI

作者: A大于_963a | 来源:发表于2020-05-02 20:11 被阅读0次

前一阵在看yolo v3 keras版本的代码,看完后自己重新训练了一个自己的数据集,但在测试时总觉得不太方便,于是尝试着在代码基础上增加一个可视化界面,可以进行图片的测试和视频的测试,由于是python新手,界面不求美观,但求基础功能能用,下面是最终的界面图:

界面左上部分显示检测的结果图片,右上半部分是选择图片和视频的按钮,其中点击视频检测时,选择mp4文件后,会弹出另一个视频实时检测的界面。

图片检测时,下面每一行显示一个检测结果,单击对应的结果后,图片上会只显示此结果的对应框,如下图:

目前这个代码基本功能可用,因此分享出来,修改的地方主要是修改了yolo_video.py和yolo.py模块。

感兴趣的童鞋可以在下面链接中下载源码(不包括.h5模型库):

https://github.com/markwentian/AI.git

相关文章

网友评论

      本文标题:yolo v3 keras版本增加图形界面UI

      本文链接:https://www.haomeiwen.com/subject/kxrlghtx.html