infer_detectron2_densepose_test.py

import logging
import cv2
from ikomia.utils.tests import run_for_test

logger = logging.getLogger(__name__)


def test(t, data_dict):
    logger.info("===== Test::infer detectron2 densepose =====")
    logger.info("----- Use default parameters")
    img = cv2.imread(data_dict["images"]["detection"]["coco"])
    input_img_0 = t.get_input(0)
    input_img_0.set_image(img)
    return run_for_test(t)