predict.py

import os
import sys
__dir__ = os.path.dirname(os.path.abspath(__file__))
sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))

import cv2

from engine import build_engine
from utils import config
from utils.get_image_list import get_image_list


def main():
    args = config.parse_args()
    config_dict = config.get_config(
        args.config, overrides=args.override, show=False)
    config_dict.profiler_options = args.profiler_options
    engine = build_engine(config_dict)

    image_list = get_image_list(config_dict["Global"]["infer_imgs"])
    for idx, image_file in enumerate(image_list):
        img = cv2.imread(image_file)[:, :, ::-1]
        input_data = {"input_image": img}
        output = engine.process(input_data)
        print(output)


if __name__ == '__main__':
    main()