mmdetection自定义取出检测结果脚本及image_demo.py解析
时间:2023-04-27 21:07:01
之所以要说image_demo.py,是因为其和video_demo.py、webcam_demo.py这三者大不小,为我们取出其他接口的测试结果提供了思路。image_demo.py在mmd\demo文件夹下。首先,我们给出image_demo.py参数分析器和main()注释方法,整个过程比较简单,然后逐一分析。
import asyncio from argparse import ArgumentParser from mmdet.apis import (async_inference_detector, inference_detector,init_detector, show_result_pyplot) def parse_args(): parser = ArgumentParser() # 创建参数解析器 parser.add_argument('img', help='Image file') # 图像需要测试 parser.add_argument('config', help='Config file') # 配置文件 parser.add_argument('checkpoint', help='Checkpoint file') # 模型的权重文件 parser.add_argument('--device', default='