26 lines
793 B
Python
26 lines
793 B
Python
from sophon_chakcy.utils.task_manager import TaskManager, TaskArgs
|
|
import time
|
|
from sophon_chakcy.sail import MultiEngine
|
|
|
|
def main():
|
|
task_manager = TaskManager(16)
|
|
|
|
for i in range(16):
|
|
task_name = f"yolo_test_{i}"
|
|
output_path = f"rtsp://192.168.137.1:8554/stream{i}"
|
|
input_path = f"./resources/datasets/test_car_person_1080P_{i}.mp4"
|
|
args = TaskArgs(
|
|
task_name=task_name,
|
|
algorithms=["yolo_example", "yolo_example", "yolo_example"],
|
|
input=input_path,
|
|
output=output_path,
|
|
dev_id=0
|
|
)
|
|
task_manager.submit(args)
|
|
start_time = time.time()
|
|
task_manager.wait_for_all()
|
|
print(time.time() - start_time)
|
|
task_manager.shutdown()
|
|
|
|
if __name__ == "__main__":
|
|
main() |