31 lines
821 B
Python
31 lines
821 B
Python
import time
|
|
from sophon_chakcy.utils.task_manager import TaskManager, TaskArgs
|
|
|
|
|
|
def main():
|
|
task_manager = TaskManager(16)
|
|
start_time = time.time()
|
|
print(start_time)
|
|
for i in range(16):
|
|
task_name = f"yolo_test_{i}"
|
|
args = TaskArgs(
|
|
task_name=task_name,
|
|
algorithms=["yolo_example", "yolo_example", "yolo_example"],
|
|
# input="rtsp://192.168.137.1:8554/stream",
|
|
input="./resources/datasets/test_car_person_1080P.mp4",
|
|
output=f"rtsp://192.168.137.1:8554/stream_{i}",
|
|
)
|
|
task_manager.submit(args)
|
|
|
|
# time.sleep(30)
|
|
|
|
# for i in range(1):
|
|
# task_name = f"yolo_test_{i}"
|
|
# task_manager.stop_task(task_name)
|
|
|
|
# print(time.time() - start_time)
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |