PaddleClas/deploy/python/ppshitu_v2/examples/tasks/image_tasks.py

13 lines
286 B
Python

import cv2
class SingleImageTask:
def __init__(self, config, engine):
self.image_path = config.get("image_path")
self.engine = engine
def run(self):
image = cv2.imread(self.image_path)
output = self.engine.process(image)
print(output)