MMYOLO、MMROTATE图片批量检测脚本

对官方给的img_demo.py做了一点改动，可以输入图片文件夹，将结果输出到output文件夹中：

# Copyright (c) OpenMMLab. All rights reserved.
from argparse import ArgumentParser

import mmcv
from mmdet.apis import inference_detector, init_detector

from mmrotate.registry import VISUALIZERS
from mmrotate.utils import register_all_modules
import os


def parse_args():
    parser = ArgumentParser()
    # parser.add_argument('img', help='Image file') # 图片
    parser.add_argument('--input_dir', help='Path to input directory') # 输入文件夹路径
    parser.add_argument('--output_dir', help='Path to output directory') # 输出文件夹路径
    parser.add_argument('config', help='Config file') # 配置文件
    parser.add_argument('checkpoint', help='Checkpoint file')