/
Alx89
/
OpenCV
Обзор
Документация
Войти
/
Alx89
/
OpenCV
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
4.x
modules/gapi/misc/python/test/test_gapi_ot.py
58 строк
2 KB
Anastasiya Pronina
Enabled VAS OT in G-API Python interface
19 дек 2023, 20:51
19 дек 2023, 20:51
d20727a
Код
Авторство
О чём код?
#!/usr/bin/env python import numpy as np import cv2 as cv import os import sys import unittest from tests_common import NewOpenCVTests try: if sys.version_info[:2] < (3, 0): raise unittest.SkipTest('Python 2.x is not supported') class gapi_ot_test(NewOpenCVTests): def test_ot_smoke(self): # Input img_path = self.find_file('cv/face/david2.jpg', [os.environ.get('OPENCV_TEST_DATA_PATH')]) in_image = cv.cvtColor(cv.imread(img_path), cv.COLOR_RGB2BGR) in_rects = [ (138, 89, 71, 64) ] in_rects_cls = [ 0 ] # G-API g_in = cv.GMat() g_in_rects = cv.GArray.Rect() g_in_rects_cls = cv.GArray.Int() delta = 0.5 g_out_rects, g_out_rects_cls, g_track_ids, g_track_sts = \ cv.gapi.ot.track(g_in, g_in_rects, g_in_rects_cls, delta) comp = cv.GComputation(cv.GIn(g_in, g_in_rects, g_in_rects_cls), cv.GOut(g_out_rects, g_out_rects_cls, g_track_ids, g_track_sts)) __, __, __, sts = comp.apply(cv.gin(in_image, in_rects, in_rects_cls), args=cv.gapi.compile_args(cv.gapi.ot.cpu.kernels())) self.assertEqual(cv.gapi.ot.NEW, sts[0]) except unittest.SkipTest as e: message = str(e) class TestSkip(unittest.TestCase): def setUp(self): self.skipTest('Skip tests: ' + message) def test_skip(): pass if __name__ == '__main__': NewOpenCVTests.bootstrap()