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

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


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

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


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

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

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

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容