之所以要说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='