代码地址:DetVisGUI

1.ReadMe文件翻译

1.1 Update

2021/04/17 : Add save all images function. (添加保存图片功能)
2021/01/05 : Support json format for det_file. (增加对json格式的支持)
2020/12/08 : Support inference model and directly show the results on GUI. (支持交互式模型并直接显示结果在GUI上)

1.2 介绍

这是轻量级的GUI用于可视化mmdetection结果,它可以显示不同阈值的检测结果,可以非常方便的验证检测结果Groundtruth
在这里插入图片描述视频和文字描述:https://www.youtube.com/watch?v=4imQyECTik0,视频中的主要命令参考下面的分支。

1.3 依赖

-- mmdetection -- tqdm

通过conda list 命令发现,安装好mmdetection之后,发现两个依赖都已经安装好。
在这里插入图片描述
在这里插入图片描述

1.4 代码

克隆这个仓库。

git clone -b mmdetection https://github.com/Chien-Hung/DetVisGUI.git
cd DetVisGUI

在DGX上可以clone但是报错,所以将其download到本地,并上传到用户文件夹。

1.5 示范案例

我采样了COCO数据集和VOC2007数据集的一小部分,并运行mmdetection获得检测结果(*.pkl),用这些文件来展示Demo。

python DetVisGUI.py ${CONFIG_FILE} [--det_file ${RESULT_FILE}] [--stage ${STAGE}] [--output ${SAVE_DIRECTORY}]

Arguments(参数):
CONFIG_FILE: Config file of mmdetction.`
Optional Arguments:(可选参数)
RESULT_FILE: Filename of the output results in pickle / json format. (输出文件结果为pickle格式或json格式)
STAGE: The stage [train / val / test] of the result file, default is ‘val’. (结果文件的三个阶段,train,val,test,默认为val)
SAVE_DIRECTORY: The directory for saving display images, default is ‘output’.(保存展示图片,默认为output文件夹)

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐